JavaRush/Помощь по Java задачам/Помогите разобраться с Collectors toMap()

33 уровень

14.10.2020
4321просмотров
12комментариев

Помогите разобраться с Collectors toMap()

Решен

На просторах интернета наткнулся на метод Collectors toMap(). С помощью стримов можно инициализировать мапу. Нашел такой пример: есть лист List<Animal> list заполненный объектами Animal.

Map<Integer, Animal> map = list.stream()
      .collect(Collectors.toMap(Animal::getId, animal -> animal));

Тут все понял, создается стрим для списка, для каждого элемента списка находится id и ложится в мапу как ключ, так же к этому ключу ложится соответствующий элемент. Вопрос: я хочу из своего листа с Integer элементами положить в мапу все элементы как ключи, как значения просто 1. Сделать я это хочу с помощью стримов.

List<Integer> list = Arrays.asList(5, 3, 7, 11, 456, 1, 2, -6);
Map<Integer, Integer> map = list.stream().collect(Collectors.toMap(i -> i, 1));

Что я делаю: создаю стрим для списка с числовыми значениями, в мапу как ключ ложу каждый элемент, как значение просто 1. Код не работает, объясните пожалуйста, как реализовать правильно?

Комментарии (12)

популярные
новые
старые

Для того, чтобы оставить комментарий Вы должны авторизоваться

hidden #598481

Уровень 27

14 октября 2020, 16:14

Еще обрати внимание, что если в твоем листе будут одинаковые числа, то выпадет исключение, т.к. ты не задал merge function. Сейчас твой код аналогичен такому циклу:

Map<Integer, Integer> map = new HashMap<>();
for (Integer i : list) {
    if (map.put(i, getOne(i)) != null) {
        throw new IllegalStateException("Duplicate key");
    }
}

Чтобы этого не происходило, у метода toMap существует третий параметр mergeFunction. Это тоже функция, но обязательно с двумя аргументами на входе. Она должна определять правила, по которому будут объединяться пары с одинаковыми ключами. Например, такой метод всегда будет выбирать имеющееся значение перед новым:

Map<Integer, Integer> map = list.stream()
                                .collect(Collectors.toMap(i -> i,
                                                          Test::getOne,
                                                          (oldValue, newValue) -> oldValue
                                ));

А такая функция объединения позволит посчитать количество всех чисел в листе:

public class Test {

    public static void main(String[] args) {
        List<Integer> list = Arrays.asList(5, 3, 7, 11, 456, 1, 2, -6, 3, 3, 3, 5, 5, 5, 5, 5);

        Map<Integer, Integer> map = list.stream()
                                        .collect(Collectors.toMap(
                                                i -> i,
                                                Test::getOne,
                                                (oldValue, newValue) -> oldValue + 1
                                        ));

        map.entrySet().forEach(System.out::println);
    }

    private static Integer getOne(int i) {
        return 1;
    }
}

Пирожок

Уровень 33

Expert

14 октября 2020, 16:27

Оу, я это не проверил, думал если дубликаты, то просто будет перезаписываться новое значение вместо старого. Очень полезно, спасибо!

Ksenia Volkova Java Developer в DXC

Master

14 октября 2020, 15:42

решение

toMap в качестве второго аргумента хочет функцию (которая в твоем случае должна принимать Integer и возвращать Integer). А ты вместо функции подсовываешь единицу. Тебе нужно что-то типа

Map<Integer, Integer> map = list.stream().collect(Collectors.toMap(i -> i, i -> 1));

Пирожок

Уровень 33

Expert

14 октября 2020, 15:48

Огромное спасибо, получилось! Прочитал в джава доках что toMap ждет две Function, но не понял что это =) буду разбираться дальше.

Ksenia Volkova Java Developer в DXC

Master

14 октября 2020, 15:51

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

Пирожок

Уровень 33

Expert

14 октября 2020, 15:54

Можно вопрос по интерфейсу Function javaDocs . Из доков Functional Interface: This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference. Либо лямбда выражение (как в моем случае), либо ссылка на метод? вот этого не понял Например:

public static void main(String[] args) {

    List<Integer> list = Arrays.asList(5, 3, 7, 11, 456, 1, 2, -6);
    //Map<Integer, Integer> map = list.stream().collect(Collectors.toMap(i -> i, getOne())); не рабочий код
    Map<Integer, Integer> map = list.stream().collect(Collectors.toMap(i -> i, i -> getOne()));

    map.entrySet().forEach(System.out::println);
}

public static Integer getOne() {
    return 1;
}

Пирожок

Уровень 33

Expert

14 октября 2020, 15:55

То есть в любом случае должно быть лямбда выражение?

Пирожок

Уровень 33

Expert

14 октября 2020, 15:57

Не заметил ваш второй комментарий, ознакомлюсь обязательно, спасибо=)

Wladyslaw Java Developer

Master

14 октября 2020, 16:00

Можно и метод референс, просто синтаксис другой. В твоём случае

ИмяКласса::getOne

Но это такое. Разберёшься с функциональными интерфейсами и лямбадми - понимание метод референсов практически само придёт

Ksenia Volkova Java Developer в DXC

Master

14 октября 2020, 16:03

Лямбда или ссылка на метод. В твоем примере с животными Animal::getId -это method reference animal -> animal - лямбда (хоть и совсем примитивная) В твоем последнем примере getOne не может использоваться в качестве method reference, потому что он не соответствует интерфейсу Function. Function должен принимать значение и возвращать значение. Чтобы использовать в качестве второго аргумента Solution::getOne , нужно чтобы этот метод принимал параметр Integer

public static Integer getOne(Integer i) {
    return 1;
}

Пирожок

Уровень 33

Expert

14 октября 2020, 16:05

Понял, спасибо за ответы!

Wladyslaw Java Developer

Master

14 октября 2020, 16:07

Упс. Да, все верно Ксения говорит. Меня не слушай, я ошибься. Глянул джавадоки.