문제

---

비어있는 공집합 S가 주어졌을 때, 아래 연산을 수행하는 프로그램을 작성하시오.

• add x: S에 x를 추가한다. (1 ≤ x ≤ 20) S에 x가 이미 있는 경우에는 연산을 무시한다.
• remove x: S에서 x를 제거한다. (1 ≤ x ≤ 20) S에 x가 없는 경우에는 연산을 무시한다.
• check x: S에 x가 있으면 1을, 없으면 0을 출력한다. (1 ≤ x ≤ 20)
• toggle x: S에 x가 있으면 x를 제거하고, 없으면 x를 추가한다. (1 ≤ x ≤ 20)
• all: S를 {1, 2, ..., 20} 으로 바꾼다.
• empty: S를 공집합으로 바꾼다.

입력

---

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.
둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

출력

---

check 연산이 주어질때마다, 결과를 출력한다.

문제풀이 (Java)

---

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;

public class Main {

    public static void add(Set<Integer> s, int n) {
        s.add(n);
    }

    public static void remove(Set<Integer> s, int n) {
        s.remove(n);
    }

    public static String check(Set<Integer> s, int n) {
        if (s.contains(n)) {
            return "1";
        } else {
            return "0";
        }
    }

    public static void toggle(Set<Integer> s, int n) {
        if (s.contains(n)) {
            s.remove(n);
        } else {
            s.add(n);
        }
    }

    public static void all(Set<Integer> s) {
        s.clear();

        for(int i = 1; i <= 20; i ++) {
            s.add(i);
        }
    }

    public static void empty(Set<Integer> s) {
        s.clear();
    }

    public static void main(String[] args) throws Exception {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());

        Set<Integer> set = new HashSet<>();
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        for(int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String command = st.nextToken();

            if(command.equals("all")) {
                all(set); continue;
            } else if(command.equals("empty")) {
                empty(set); continue;
            }

            int num = Integer.parseInt(st.nextToken());

            if(command.equals("add")) {
                add(set, num);
            } else if(command.equals("remove")) {
                remove(set, num);
            } else if(command.equals("check")) {
                bw.write(check(set, num));
                bw.newLine();
            } else if(command.equals("toggle")) {
                toggle(set, num);
            }
        }

        bw.flush();
        bw.close();

    }

}

처음에 BufferedWriter로 출력하지 않고 System.out.println()을 사용했더니 시간 초과가 나왔다.
앞으로 문제 풀때는 입력에는 BufferedReader, 출력에는 BufferedWriter를 사용하는 습관을 들여야겠다.