设计哈希映射

/*
 * 思路：
 * 1. 使用一个ArrayList来存储键值对，每个键值对用一个数组表示。
 * 2. put方法检查是否已有该键，如果有则更新值，如果没有则添加新的键值对。
 * 3. get方法遍历列表查找键并返回对应值，如果没有找到则返回-1。
 * 4. remove方法遍历列表找到键并将其移除。
 * 5. 使用Stream API简化一些操作。
 */

import java.util.ArrayList;
import java.util.List;

public class MyHashMap {
    private List<int[]> map;

    /** Initialize your data structure here. */
    public MyHashMap() {
        map = new ArrayList<>();
    }

    /** value will always be non-negative. */
    public void put(int key, int value) {
        for (int[] pair : map) {
            if (pair[0] == key) {
                pair[1] = value;
                return;
            }
        }
        map.add(new int[]{key, value});
    }

    /** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */
    public int get(int key) {
        return map.stream()
                .filter(pair -> pair[0] == key)
                .map(pair -> pair[1])
                .findFirst()
                .orElse(-1);
    }

    /** Removes the mapping of the specified value key if this map contains a mapping for the key */
    public void remove(int key) {
        map.removeIf(pair -> pair[0] == key);
    }
}