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

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

HashMaps также поддерживают удаление и обновление записей с помощью методов 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();

        // Преобразование строки 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));
    }
}