过滤仅具有相交数据的 ArrayList

Question

我有以下课程：

public class School{
    List<ClassRoom> classRooms;
}

public class ClassRoom{
    List<Student> students;
}

public class Student{
    String name;
    List<Long> typeIdList;
}

我只需要得到typeId，这是给定教室S内所有学生的共同因素。

为了能够让 typeID=123 的给定教室 S 中的所有学生，我执行以下操作：

final long typeIdToSearchFor = ...;
Collection<Student> filtered = Collections2.filter(students,
    new Predicate<Student>() {
        @Override
        public boolean apply(Student s) {
            return s.typeId == typeIdToSearchFor;
        }
    }
);

只是想知道 Guava 是否可以处理这样的事情？ 我所说的交集是指在所有情况下都必须考虑这种类型。

我知道for 循环更具可读性，但我只是发现了 Guava 功能。

Answer 1

您可以使用 Multiset 来计算出现次数：

ClassRoom classRoom = /* comes from somewhere */;
List<Student> students = classRoom.getStudents();

// Aggregate all the typeIds.
Multiset<Long> typeIds = HashMultiset.create();
for (Student student : students) {
    // Assuming a student doesn't have duplicated typeIds:
    typeIds.addAll(student.getTypeIds());
}

// Find which typeIds are present for all the students.
for (Multiset.Entry<Long> entry : typeIds.entrySet()) {
    if (entry.getCount() == students.size()) {
        System.out.println(entry.getElement());
    }
}

Answer 2

如果您要检查所有Students 是否具有特定的typeId，请将Iterables.all 与现有谓词一起使用。

您还可以使用Multimaps.index() 创建由typeId 索引的Students 的Multimap：

Multimap<Long, Student> studentsByTypeId = Multimaps.index(students, new Function<Student, Long>() {
  public Long apply(Student s) {
    return s.typeId;
  }
};

然后你可以使用studentsByTypeId.keySet()获得唯一的typeIds。

您可以使用studentsByTypeId.keySet().size() == 1 检查它们是否都共享相同的typeId。

Answer 3

如果您想要相交的那些，则不应与特定的进行比较，您至少应该检查它是否包含在另一个集合的 id 中。像这样：

new Predicate<Student>() {
    @Override
    public boolean apply(Student s) {
        return otherIds.contains(s.typeId);
    }
}

但我仍然认为，如果您对两个集合都执行二进制搜索（在对它们进行排序之后），您可以更快地得到答案。像这样的

Collections.sort(list1);
Collections.sort(list2);

List<E> intersected = new ArrayList<E>();

for(E element : list1){
    if(Collections.binarySearch(list2, element) >= 0){
        intersected.add(element);
    }
}

您甚至可以找到最小的列表。它可以帮助它获得一些性能。