给定已排序的Integer ArrayLists的ArrayList事务,我正在编写代码以返回其唯一元素.例如,给定
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中,LinkedHashSet通过其定义来保持排序并删除重复项.
Set<Integer> uniqEl = new LinkedHashSet<>();
for (List<Integer> l : transactions) {
for (Integer n : l) {
uniqEl.add(n);
}
}
虽然我的代码通过利用Java库来完成工作,但我希望实现更高效的实现.是否有更好的算法来从列表列表中生成排序的唯一元素列表?
解决方法:
使用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));
使用此解决方案,您可以并行运行它,但您必须衡量它是否可以提高性能.