[BOJ] 11723 - 집합 (🥈 실버 5티어)

사용 언어: Java

문제

집합

풀이

나의 풀이

자바

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.BufferedWriter;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter out = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(in.readLine());
        ArrayList<Integer> lst = new ArrayList<>();
        StringTokenizer stk;

        for (int i=0; i<n; i++) {
            String command = in.readLine();
            if (command.equals("all")) {
                lst.clear();
                for (int j=1; j<21; j++) {
                    lst.add(j);
                }
            } else if (command.equals("empty")) {
                lst.clear();
            } else {
                stk = new StringTokenizer(command);
                String cmd = stk.nextToken();
                int num = Integer.parseInt(stk.nextToken());
                
                if (cmd.equals("add")) {
                    if (!lst.contains(num)) lst.add(num);
                } else if (cmd.equals("remove")) {
                    if (lst.contains(num)) lst.remove(Integer.valueOf(num)); // 그냥 remove(num)은 idx가 num인 요소를 삭제한다
                } else if (cmd.equals("check")) {
                    if (lst.contains(num)) out.write(1 + "\n");
                    else out.write(0 + "\n");
                } else if (cmd.equals("toggle")) {
                    if (lst.contains(num)) lst.remove(Integer.valueOf(num));
                    else lst.add(num);
                }
            }
        }
        out.close();
        in.close();
    }
}

• 테스트 케이스: 성공
• 제출 결과: 성공

다른 풀이

자바

참고

---

💛 개인 공부 기록용 블로그입니다. 👻

맨 위로 이동하기