[Java] BOJ 11723 집합

문제 

문제 링크 https://www.acmicpc.net/problem/11723

• 비어있는 공집합 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를 공집합으로 바꾼다.
• check 연산이 주어질때마다, 결과를 출력하기
  

아이디어

• 연산 횟수만큼 boolean 배열을 만들어서 true, false로 값을 관리하면 일일이 값을 넣었다가 빼는 것 보다 좀 더 효율적이지 않을까라는 생각이 들었다. 
• if문을 활용하여 연산에 따라 다른 상호작용을 하면 될 것 같았다. 
  
  

겪은 시행착오

•  바보 같게도... "add 1" 처럼 모두 "문자 + 숫자" 형태로 입력이 들어오는 것이 아닌데 항상 연산 방법을 입력받고 nextToken을 써서 문자열과 숫자를 저장하려다보니 오류가 났다.  (all이나 empty는 뒤에 숫자가 없다...)
• 항상 예외처리 할게 없는지 유념하자....!

앞으로도 파이팅!!

 

코드

import  java.io.*;
import java.util.*;

public class BOJ11723 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int n = Integer.parseInt(br.readLine());

        boolean arr[] = new boolean[21];

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

            String cal = st.nextToken();

            if(cal.equals("add")){
                int x = Integer.parseInt(st.nextToken());

                if(arr[x] == false){
                    arr[x] = true;
                }
            }
            if(cal.equals("remove")){
                int x = Integer.parseInt(st.nextToken());

                if(arr[x] == true){
                    arr[x] = false;
                }
            }
            if(cal.equals("check")){
                int x = Integer.parseInt(st.nextToken());

                if(arr[x] == true){
                    sb.append(1 + "\n");
                }
                else{
                    sb.append(0 + "\n");
                }
            }
            if(cal.equals("toggle")){
                int x = Integer.parseInt(st.nextToken());

                if(arr[x] == true){
                    arr[x] = false;
                }
                else{
                    arr[x] = true;
                }
            }
            if(cal.equals("all")){
                for(int j = 0; j < 21; j++){
                    arr[j] = true;
                }
            }
            if(cal.equals("empty")){
                for(int j = 0; j < 21; j++){
                    arr[j] = false;
                }
            }
        }

        System.out.println(sb);
    }
}