2016-02-15 3 views
3

Учитывая ArrayList transactions отсортированных целых массивов ArrayLists, я пишу код для возврата его уникальных элементов. Например, приСписок уникальных элементов, заданных списком списков

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, я хочу более эффективную реализацию. Любые идеи для лучшего алгоритма для создания отсортированного списка уникальных элементов из списка списков?

+2

Более эффективный? Итак, вы действительно измерили и это реальное узкое место в вашей заявке? (Заметьте, почему бы не использовать «TreeSet»?) – Tunaki

+1

'uniqEl.addAll (l)' может быть * незначительно * более эффективным. –

+0

Ваш результат будет «1, 2, 3, 5, 8, 13, 21, 6, 10, 11', который не является порядком сортировки, который вы хотели. Использование «TreeMap», например, предложенного @Tunaki, даст вам желаемый результат. – Andreas

ответ

3

Вы не сможете иметь что-то более эффективным, что с помощью TreeSet и добавление все списки в этот набор. A 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 потоков на одной линии, что:

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

С помощью этого решения, вы можете запустить его параллельно, но вы должны измерить, что улучшает производительность.

2

Если на «более эффективный», вы имеете в виду более компактным, то этот функциональный подход может сделать трюк для вас:

List<Integer> list = 
transactions.stream() 
      .flatMap(List::stream) 
      .distinct() 
      .sorted() 
      .collect(Collectors.toList()); 
+0

Вместо 'l -> l.stream()' не будет ли ссылка метода «Список :: поток» более компактной? – Andreas

+0

@Andreas: Вы выигрываете –

2

С Eclipse Collections должно работать:

MutableList<MutableList<Integer>> transactions = 
    Lists.mutable.with(
     Lists.mutable.with(1, 1, 2, 3, 5, 8, 13, 21), 
     Lists.mutable.with(2, 3, 6, 10), 
     Lists.mutable.with(11, 21)); 
SortedSet<Integer> flattenedDistinct = 
    transactions.asLazy().flatCollect(l -> l).toSortedSet(); 

Assert.assertEquals(
    SortedSets.mutable.of(1, 2, 3, 5, 6, 8, 10, 11, 13, 21), flattenedDistinct); 

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

SortedSet<Integer> flattenedDistinct = 
    transactions.flatCollect(l -> l, SortedSets.mutable.empty()); 

Чтобы избежать бокса ints вы можете использовать примитивные коллекции, доступные в коллекциях Eclipse.

MutableList<MutableIntList> transactions = Lists.mutable.with(
    IntLists.mutable.with(1, 1, 2, 3, 5, 8, 13, 21), 
    IntLists.mutable.with(2, 3, 6, 10), 
    IntLists.mutable.with(11, 21)); 
MutableIntList flattenedDistinct = 
    transactions.injectInto(IntSets.mutable.empty(), MutableIntSet::withAll).toSortedList(); 

Assert.assertEquals(
    IntLists.mutable.with(1, 2, 3, 5, 6, 8, 10, 11, 13, 21), flattenedDistinct); 

Примечание: Я являюсь автором коллекций Eclipse.

Смежные вопросы