题解 | 【模板】多重集合操作

要注意，使用Map保存键值的时候，新增某个数，数量值记得加上原来的数值。删除某个数，是只删除一个，所以要先判断该数是否存在，然后更新数量。然后求集合的大小，这里应该是求的所有元素的值得数量总和，因此不能用size()，而是维护一个值的总数totalSize。

import java.util.*;

public class Main {
    // 静态变量
    static TreeMap<Integer, Integer> tm = new TreeMap<>();
    static int totalSize = 0;
    public static void insertValue(int x) {
        // TODO: 实现插入逻辑
        // 如果存在则是原来的数量+1
        // tm.putIfAbsent(x, 1); // 默认值为1
        tm.put(x, tm.getOrDefault(x, 0) + 1);
        totalSize++;
    }

    public static void eraseValue(int x) {
        // TODO: 实现删除逻辑
        // tm.remove(x); // 按照键删除
        // 删除只删除一个
        if (tm.containsKey(x)){
            int cnt = tm.get(x);
            if (cnt > 1) {
                tm.put(x, cnt - 1);
            }else {
                tm.remove(x);
            }
            totalSize--;
        }
    }

    public static int xCount(int x) {
        // TODO: 求x在集合中的个数
        return tm.getOrDefault(x, 0); // 默认不存在返回个数0
    }

    public static int sizeOfSet() {
        // TODO: 返回集合大小
        return totalSize;
    }

    public static int getPre(int x) {
        // TODO: 实现找前驱
        Integer pre = tm.lowerKey(x);
        return pre != null ? pre : -1;
    }

    public static int getBack(int x) {
        // TODO: 实现找后继
        // 使用导航方法即可，注意判断
        Integer back = tm.higherKey(x); 
        return back != null ? back : -1;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int q = scanner.nextInt();
        while (q-- > 0) {
            int op = scanner.nextInt();
            switch (op) {
                case 1:
                    insertValue(scanner.nextInt());
                    break;
                case 2:
                    eraseValue(scanner.nextInt());
                    break;
                case 3:
                    System.out.println(xCount(scanner.nextInt()));
                    break;
                case 4:
                    System.out.println(sizeOfSet());
                    break;
                case 5:
                    System.out.println(getPre(scanner.nextInt()));
                    break;
                case 6:
                    System.out.println(getBack(scanner.nextInt()));
                    break;
            }
        }
        scanner.close();
    }
}