Блог

Главная - Вопросы по программированию - Группировка только последовательных элементов на основе свойства с использованием потоков Java

Группировка только последовательных элементов на основе свойства с использованием потоков Java

#java #java-stream

#java #java-stream

Вопрос:

У меня есть список объектов, каждый объект имеет идентификатор. Я хочу группировать элементы по их идентификатору, учитывая, что они являются последовательными. Например, если объекты:

 (id1, id1, id1, id2, id2, id3, id3, id2, id2, id4)

затем группы должны быть:

 (id1, id1, id1), (id2, id2), (id3, id3), (id2, id2), (id4)

Может ли это быть достигнуто с помощью Java Streams API?

Комментарии:

1. Ожидаемый результат — List<List<ObjectName>> или Map<Long, List<ObjectName>>

2. Подсказка: objects.stream().collect(Collectors.groupingBy(ObjectName::getId))

3. Но это сгруппировало бы все элементы с одинаковым идентификатором, даже если они не являются последовательными. Я хочу, чтобы они группировались, только если они последовательны и имеют одинаковый идентификатор.

4. Ах, извините, я неправильно понял ваш вопрос, в этом случае, я думаю, использование stream может быть плохим, простой цикл for может выполнить эту работу за вас.

5. Почему это понижение?

Ответ №1:

Это очень возможно с помощью изменяемого сокращения, приводящего к List<List<String>> . Для лучшей читаемости я рекомендую разделить на методы:

 List<List<String>> newList = list.stream().collect(
    ArrayList::new,
    (lists, string) -> {
        if (lists.isEmpty()) {
            withNewList(lists, string);
        } else {
            withNewString(lists, string);
        }
    },
    ArrayList::addAll
);
 
 // adds a new inner list with a single item (string)
static void withNewList(ArrayList<List<String>> lists, String string) {
    List<String> newList1 = new ArrayList<>();
    newList1.add(string);
    lists.add(newList1);
}
 
 static void withNewString(ArrayList<List<String>> lists, String string) {
    // if the last inserted list has a same item
    List<String> lastList = lists.get(lists.size() - 1);   
    if (lastList.contains(string)) {
        // append it to the last inner list
        lastList.add(string);
    } else {
        // or else create a new list with a single item (string)
        withNewList(lists, string);
    }
}

Учитывая следующие list входные данные:

 List<String> list = List.of(
    "id1", "id1", "id1", "id2", "id2", "id3", "id3", "id2", "id2", "id4");

… когда вы распечатываете результат, вывод выглядит следующим образом:

[[id1, id1, id1], [id2, id2], [id3, id3], [id2, id2], [id4]]

Ответ №2:

Для удобства навигации к последнему (предыдущему) элементу я бы использовал LinkedList из LinkedLists:

 public static void main(String[] args) {
    ...
    Stream.of(id1, id1, id1, id2, id2, id3, id3, id2, id2, id4)
            .sequential() // order is essential
            .collect(LinkedList::new, (listOfLists, object) -> {
        if (listOfLists.isEmpty() || listOfLists.getLast().getLast() != object) {
            listOfLists.add(new LinkedList<>(List.of(object)));
        } else {
            listOfLists.getLast().add(object);
        }
    }, List::addAll);
    ...
}
Метки: