Создание подмножества набора в Java

У меня есть LinkedHashSet, i.e упорядоченный набор. Я пытаюсь найти функцию, чтобы просто вернуть подмножество набора, i.e первые 20 элементов множества. Я знаю, что могу сделать это, создав новый набор, а затем заполнив его итерацией первого набора, но я надеялся на что-то более сжатое.

также взглянул на библиотеки гуавы Google, но не смог увидеть, что я хотел.

19
java linkedhashset set subset

6 ответов

В Гуава:

Set<Integer> subset = ImmutableSet.copyOf(Iterables.limit(set, 20));

отметим, что Iterables.limit() оценивается лениво, поэтому создается только одна дополнительная коллекция.

33

вы могли бы сделать это:

Set<Integer> set = new LinkedHashSet<>();
for (int i = 0; i < 50; i++) {
   set.add(i);
}

List<Integer> list = new ArrayList<>(set);
Set<Integer> subSet = new LinkedHashSet<>(list.subList(0, 20));
15

решение с использованием потоков и коллекторов из Java 8:

Set<Integer> subSet = set.stream()
    .limit(20)
    .collect(toCollection(LinkedHashSet::new));
    // You could also collect to something else 
    // with another collector like this: 
    // .collect(toList());

предполагается import static java.util.stream.Collectors.toCollection;.

9

в Java 8, вы можете сделать

youSet.stream()
   .skip(start) // the offset
   .limit(count) // how many items you want
   .collect(Collectors.toSet());
7

вы можете использовать сначала a SortedSet Как subSet метод.

вы также можете добавить содержимое своего набора в List и с помощью subList метод. Но это зависит от объема данных, хранящихся в Set как вы не хотели бы дублировать огромный объем данных.

в противном случае вы должны оставаться с итерацией по набору, поскольку это будет более эффективно.

4

простой вспомогательный метод (вы можете использовать его для установки или любой другой коллекции):

public static <T> List<T> listOf(final Collection<T> set, final int limit) {
    final List<T> list = new ArrayList<>(limit);

    final Iterator<T> i = set.iterator();
    for (int j = 0; j < limit && i.hasNext(); j++) {
        list.add(i.next());
    }

    return list;
}
1