使用Collections.min / max查找多个最小值/最大值

时间:2018-04-02 07:58:39

标签: java collections min

在我正在研究的一个例子中,我试图找到列表中最小的三个元素(不对列表进行排序),然后将这三个元素添加到一个新列表中。

因为这是一个例子,我可以简单地使用for循环,使用Collections.min(list)将该元素添加到新列表,然后从原始列表中删除该元素。如果我没有删除元素,我将获得相同的元素三次。然而,通过删除元素,我得到了我想要的结果。

如何在不删除列表中的max / min元素的情况下执行此操作?

2 个答案:

答案 0 :(得分:1)

如果你想找到最多/最少3个元素,我建议你使用PriorityQueue

PriorityQueue<Integer> pq = new PriorityQueue<>(k);

然后在循环中,将元素添加到此队列。

然后,您可以通过从队列中删除并将这3个元素添加到列表中,只需从方法返回。

更好的是使用3个单独的变量并直接在主列表上循环。 注意:如果您稍后将更新3更新为某个其他值,则此方法将不可行。而PriorityQueue方法将是灵活的。

public static void main (String[] args) throws java.lang.Exception {
    Integer arr[] = {2, 6, 5, 3, 7, 9, 12, 35, 1, 3};
    List<Integer> incoming = Arrays.asList(arr);
    Comparator<Integer> maxFirstComparator = (x, y) -> Integer.compare(y,x);
    printList(getMinOrMaxKNumbers(incoming, 3, null));
    System.out.println();
    printList(getMinOrMaxKNumbers(incoming, 3, maxFirstComparator));
}

/*
 * gets the max/min K elements from the List
 * @param comparator    if null is passed the method uses natural ordering
 *
 */
private static List<Integer> getMinOrMaxKNumbers(List<Integer> incoming, int k, Comparator<Integer> comparator) {
    int n = incoming.size();
    PriorityQueue<Integer> pq = comparator == null ? new PriorityQueue<>(n) : new PriorityQueue<>(n, comparator);

    for (int i : incoming) {
        pq.add(i);
    }

    List<Integer> outgoing = new ArrayList<>(k);
    for (int i = 0; i < k; i++) {
        outgoing.add(pq.poll());
    }

    return outgoing;
}

private static void printList(List<Integer> list) {
    list.stream().forEach(x -> System.out.print(x + " "));
}

答案 1 :(得分:0)

我不认为使用标准库有任何直接的方法。

以下代码包含一个实用程序方法,用于获取一组最大元素。

它的工作原理是将最大元素保持在// Define the full struct type schema to check if nested field exists. val structToCheck = new StructField("value", new StructType().add("version",new StructType().add("major",StringType).add("minor",StringType).add("vnum",StringType))) val SQL = """ SELECT value.version.major, value.version.minor,""" + if (parquetFile.schema.contains(structToCheck)) "value.version.vnum" else "'' as vnum" + "FROM OUT_TABLE LIMIT 10" 中,只有当元素属于最大元素时才会插入元素。树集合非常适合这里,因为它可以快速找到最小元素,并测试元素是否包含在集合中。

通过这种方式,您可以获得良好的性能,并且可以灵活地使用您想要的最大元素数量。

TreeSet

在问题文本中,您不会写任何有关如何处理输入列表中的重复元素的内容。此方法返回最大的唯一元素。

如果应保留重复项,则可以使用基于树的Guava多集而不是普通public class MultipleMaxElements { public static void main(String[] args) { List<Integer> l = List.of(4, 2, 5, 8, 2, 8, 0, 1); System.out.println(maxElements(l, 3, Comparator.naturalOrder())); System.out.println(maxElements(l, 3, Comparator.<Integer>naturalOrder().reversed())); } public static <T> Set<T> maxElements(Iterable<T> list, int nrElems, Comparator<T> cmp) { TreeSet<T> maxSet = new TreeSet<>(cmp); for (T elem : list) { if (maxSet.size() < nrElems) { maxSet.add(elem); } else if (!maxSet.contains(elem) && cmp.compare(elem, maxSet.first()) > 0) { maxSet.pollFirst(); maxSet.add(elem); } } return maxSet; } }