Изменение словарей

Часть раздела Логика и управление потоком путешествия по Java на Coddy. Урок 14 из 59.

HashMap также поддерживают удаление и обновление записей с помощью методов put(), remove() и replace().

Например, сначала заполним HashMap:

HashMap<String, String> users = new HashMap<>();
users.put("alice", "Admin");
users.put("bob", "User");

Чтобы удалить ключ:

users.remove("bob"); // Удаляет "bob"
System.out.println(users);
// {alice=Admin}

Чтобы обновить ключ:

users.replace("alice", "Guest");
// Заменяет значение для alice на Guest

users.replace("Charlie", "Admin");
// Ничего не делает (ключа "Charlie" не существует)

Задание

Легко

Напиши метод с именем modifyMap, который изменяет заданный HashMap в соответствии с определёнными условиями.

Принимает HashMap с именем data, где:
- Ключи имеют тип String.
- Значения имеют тип Integer.
Принимает String key.
Принимает Integer newValue.

Метод должен вернуть обновлённый HashMap, следуя таким правилам:

Если ключ существует в data и имеет то же значение, что и newValue, используй replace(), но увеличь значение на 1.
Если ключ существует, но имеет другое значение, удали его.
Если ключ не существует, добавь его со значением newValue.

Попробуйте сами

// --- Модули для преобразования строки с hashmap в hashmap ---
import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import java.lang.reflect.Type;
// -----------------------------
import java.util.HashMap;
import java.util.Scanner;

public class Main {
    public static HashMap<String, Integer> modifyMap(HashMap<String, Integer> data, String key, int newValue) {
        // Напишите здесь свой код
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String hashMapString = scanner.nextLine();
        String key = scanner.nextLine();
        int newValue = scanner.nextInt();

        // Преобразовать String с HashMap в HashMap
        Type mapType = new TypeToken<HashMap<String, Integer>>(){}.getType();
        HashMap<String, Integer> data = new Gson().fromJson(hashMapString, mapType);

        HashMap<String, Integer> result = modifyMap(data, key, newValue);
        System.out.println(new Gson().toJson(result));
    }
}