设计数字容器系统

/*
 * 思路：
 * 1. 使用两个映射来实现：一个映射indexToNumber记录每个下标对应的数字；另一个映射numberToIndices记录每个数字对应的所有下标。
 * 2. change方法将数字插入或替换到指定下标，并相应更新两个映射。
 * 3. find方法返回给定数字的最小下标。
 * 4. 使用Java Stream API简化find方法。
 */
import java.util.*;
import java.util.stream.*;

public class NumberContainers {
    private Map<Integer, Integer> indexToNumber;
    private Map<Integer, TreeSet<Integer>> numberToIndices;

    public NumberContainers() {
        indexToNumber = new HashMap<>();
        numberToIndices = new HashMap<>();
    }

    public void change(int index, int number) {
        if (indexToNumber.containsKey(index)) {
            int oldNumber = indexToNumber.get(index);
            if (numberToIndices.containsKey(oldNumber)) {
                numberToIndices.get(oldNumber).remove(index);
                if (numberToIndices.get(oldNumber).isEmpty()) {
                    numberToIndices.remove(oldNumber);
                }
            }
        }
        indexToNumber.put(index, number);
        numberToIndices.putIfAbsent(number, new TreeSet<>());
        numberToIndices.get(number).add(index);
    }

    public int find(int number) {
        return numberToIndices.getOrDefault(number, new TreeSet<>()).stream().findFirst().orElse(-1);
    }
}