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

mol*_*und 11 java lambda java-8 java-stream

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

我认为解决方案应该是一成不变的。因此,我无法创建一个薪水最低的变量,而只能检查每个薪水是否小于或等于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);
Run Code Online (Sandbox Code Playgroud)

Rav*_*ala 13

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

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

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

仅仅遵循了这个很好的答案,我想到了编写一个符合我们目的的定制收集器。该收集器仅对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;
        }
    });
}
Run Code Online (Sandbox Code Playgroud)

这是您的客户代码。

Collection<Employee> empsWithLowestSalary = employees.stream()
                .collect(minList(Comparator.comparing(Employee::getSalary)));
Run Code Online (Sandbox Code Playgroud)

  • 该解决方案的复杂性是什么?我相信它是O(n * logn)。 (7认同)
  • 好答案。除了时间上的复杂性,我还担心的是,此解决方案将为groupingBy收集器占用的额外空间。 (3认同)