java - 使用流检索薪水最低的员工列表

This question already has answers here:

How to force max to return ALL maximum values in a Java Stream?

(8个答案)


1年前关闭。




我正在尝试从员工列表中检索薪水最低的列表。到目前为止，我设法找到了薪水最低的员工，但是如果几个员工的薪水相同，我想检索多个员工。

我认为解决方案应该是一成不变的。因此，我无法创建一个具有最低薪水的变量，而只能检查每个人的“薪水是否小于或等于Low_Salary”。我尝试了一下，它奏效了。

因此，我相信我的问题是“.min(比较器)”仅检索最低的一个。

Comparator<Employee> comparator = Comparator.comparing( Employee :: getSalary);
List<Employee> lowSalary = employees.stream()
               .min(comparator)
               .stream()
               .collect(Collectors.toList());
lowSalary.forEach(System.out::println);

最佳答案

首先创建一个TreeMap，其键是薪水。 TreeMap按其键对条目进行排序。然后获取第一个条目，即薪水最低的条目，并掌握与此相关的值。此解决方案仅对列表进行一次迭代。这是它的外观。

List<Employee> empsWithLowestSalary = employees.stream()
    .collect(Collectors.groupingBy(Employee::getSalary, TreeMap::new, Collectors.toList()))
    .firstEntry()
    .getValue();

TreeMap将 map 元素存储在红黑树中。红黑树中一个元素的插入成本为O(Log (n))。由于我们要插入n元素，因此此解决方案的总时间复杂度为O(n Log (n))。对于firstEntry()，它需要固定时间O(1)，因为它维护着分别指向树中最左边和最右边叶子节点的指针。最左边的节点代表树中的最小值，而最右边的叶子节点代表树的最大值。

仅仅遵循这个great answer，我就想编写一个满足我们目的的自定义收集器。该收集器仅对List进行一次迭代，其运行时复杂度为O(n)，明显优于上述方法。此外，它允许您在一个语句中编写客户代码。这是它的外观。

static <T> Collector<T, ?, List<T>> minList(Comparator<? super T> comp) {
    return Collector.of(ArrayList::new, (list, t) -> {
        int c;
        if (list.isEmpty() || (c = comp.compare(t, list.get(0))) == 0)
            list.add(t);
        else if (c < 0) {
            /*
             * We have found a smaller element than what we already have. Clear the list and
             * add this smallest element to it.
             */
            list.clear();
            list.add(t);
        }
    }, (list1, list2) -> {
        if (comp.compare(list1.get(0), list2.get(0)) < 0)
            return list1;
        else if (comp.compare(list1.get(0), list2.get(0)) > 0)
            return list2;
        else {
            list1.addAll(list2);
            return list1;
        }
    });
}

这是您的客户代码。

Collection<Employee> empsWithLowestSalary = employees.stream()
                .collect(minList(Comparator.comparing(Employee::getSalary)));

关于java - 使用流检索薪水最低的员工列表，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/57923706/