[백준] 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를 공집합으로 바꾼다. 

 

입력

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.

둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

 

출력

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

 

예제 입력

26
add 1
add 2
check 1
check 2
check 3
remove 2
check 1
check 2
toggle 3
check 1
check 2
check 3
check 4
all
check 10
check 20
toggle 10
remove 20
check 10
check 20
empty
check 1
toggle 1
check 1
toggle 1
check 1

 

예제 출력

1
1
0
1
0
1
0
1
0
1
1
0
0
0
1
0

 

내 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class Main_11723 {
	static int M;
	static List<Integer> list = new ArrayList<>();
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		M = Integer.parseInt(br.readLine());
		StringBuilder sb = new StringBuilder();
		
		for (int i = 0; i < M; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine(), " ");
			String oper = st.nextToken();

			int num = 0;
			here:
			switch(oper) {
			case "add":
				num = Integer.parseInt(st.nextToken());
				list.add(num);
				break;
			case "remove":
				num = Integer.parseInt(st.nextToken());
				for(int j=0; j<list.size(); j++) {
					if(list.get(j) == num) list.remove(j);
				}
				break;
			case "check":
				num = Integer.parseInt(st.nextToken());
				if(list.contains(num))
					sb.append(1).append("\n");
				else
					sb.append(0).append("\n");
				break;
			case "toggle":
				num = Integer.parseInt(st.nextToken());
				for(int j=0; j<list.size(); j++) {
					if(list.get(j) == num) {
						list.remove(j);
						break here;
					}
				}
				list.add(num);
				break;
			case "all":
				list.clear();
				for(int j=1; j<=20; j++)
					list.add(j);
				break;
			case "empty":
				list.clear();
				break;				
			}
		}

		System.out.println(sb);
		
		br.close();
	}
}

 

다 푼 후 알고리즘 분류를 보니 비트마스킹이 있었다.

 

다른 사람 코드

import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		int set = 0;
		Scanner sc = new Scanner(System.in);
		StringBuilder sb = new StringBuilder();
		for(int i=sc.nextInt()-1;i>=0;i--) {
			switch(sc.next()) {
			case "add":
				set |= 1 << sc.nextInt();
				break;
			case "check":
				sb.append((set & 1 <<sc.nextInt())!=0 ? 1 : 0).append("\n");
				break;
			case "remove":
				set &= ~(1 << sc.nextInt());
				break;
			case "toggle":
				set ^= 1 << sc.nextInt();
				break;
			case "all":
				set = (1<<21)-1;
				break;
			case "empty":
				set = 0;
				break;
			}
		}
		System.out.println(sb.toString());
	}

}