Algorithm
[백준 11723번/자바] 집합
도도고영
2023. 8. 7. 16:22
문제
비어있는 공집합 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 연산에 '&'연산자가 잘 이해가 되지않았다. gpt한테 물어보고 이해완료!
case "check":
result.append((bitset & (1 << (num - 1))) != 0? "1\n" : "0\n");
그리고 공집합으로 만드는 부분에 나는 bitset = 0으로 코드를 적었는데 참고한 블로그는 다른 표현을 썼다. 궁금해서 이 부분도 gpt한테 물어봄.
결과 코드!
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class BJ11723_2 {
public static void main(String[] args) throws NumberFormatException, IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int m = Integer.parseInt(br.readLine());
StringBuilder result = new StringBuilder();
int bitset = 0;
for (int i = 0; i < m; i++) {
StringTokenizer st = new StringTokenizer(br.readLine(), " ");
String command = st.nextToken();
int num = 0;
if(command.equals("add") || command.equals("remove") || command.equals("check") || command.equals("toggle")) {
num = Integer.parseInt(st.nextToken());
}
switch (command) {
case "add":
bitset |= (1 << (num - 1));
break;
case "remove":
bitset &= ~(1 << (num - 1));
break;
case "check":
result.append((bitset & (1 << (num - 1))) != 0? "1\n" : "0\n");
break;
case "toggle":
bitset ^= (1 << (num - 1));
break;
case "all":
bitset |= (~0);
break;
case "empty":
bitset = 0;
break;
}
}
System.out.println(result);
}
}
그리고 드디어 정답-!