题解 | 【模板】集合操作

import java.util.*;

public class Main {
    static TreeSet<Integer> set = new TreeSet<>();

    public static void insertValue(int x) {
        // TODO: 实现插入逻辑
        set.add(x);
    }

    public static void eraseValue(int x) {
        // TODO: 实现删除逻辑
        if(set.contains(x))set.remove(Integer.valueOf(x));
    }

    public static boolean xInSet(int x) {
        // TODO: 实现存在性检查
        return set.contains(x);
    }

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

    public static int getPre(int x) {
        // TODO: 实现找前驱
       return set.lower(x)==null?-1:set.lower(x);
    }

    public static int getBack(int x) {
        return set.higher(x)==null?-1:set.higher(x);
    }

    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(xInSet(scanner.nextInt()) ? "YES" : "NO");
                    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();
    }
}