java - 给定列表列表的唯一元素列表

Question

给定一个已排序的整数 ArrayList 的 ArrayList transactions，我正在编写代码以返回其唯一元素。例如，给定

transactions = [
  [1, 1, 2, 3, 5, 8, 13, 21],
  [2, 3, 6, 10],
  [11, 21]
]

我的代码应该返回唯一元素，保留排序顺序:

[1, 2, 3, 5, 6, 8, 10, 11, 13, 21]

为此，我只是将每个列表中的每个元素添加到 LinkedHashSet，根据其定义，它会保持排序并删除重复项。

Set<Integer> uniqEl = new LinkedHashSet<>();

for (List<Integer> l : transactions) {
    for (Integer n : l) {
        uniqEl.add(n);
    }
}

虽然我的代码通过利用 Java 库来完成工作，但我想要更高效的实现。对于更好的算法从列表的列表中生成排序的唯一元素列表有什么想法吗？

Answer 1

没有比使用 TreeSet 并将所有列表添加到该集合中更有效的方法了。 TreeSet 将按元素的自然升序对元素进行排序，并忽略重复元素。

public static void main(String[] args) {
    List<List<Integer>> transactions = Arrays.asList(Arrays.asList(1, 1, 2, 3, 5, 8, 13, 21), Arrays.asList(2, 3, 6, 10), Arrays.asList(11, 21));

    SortedSet<Integer> set = new TreeSet<>();
    for (List<Integer> l : transactions) {
        set.addAll(l);
    }
}

当然，您可以使用 Java 8 Streams 来一行:

SortedSet<Integer> set = transactions.stream()
                                     .flatMap(List::stream)
                                     .collect(Collectors.toCollection(TreeSet::new));

使用此解决方案，您可以并行运行它，但您必须衡量它是否提高了性能。