티스토리 뷰

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 연산이 주어질때마다, 결과를 출력한다.


https://dragon-h.tistory.com/28

https://travelbeeee.tistory.com/451

두 블로그를 참고해서 풀었다. 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);
		
		
	}
}

그리고 드디어 정답-!

공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/04   »
1 2 3 4 5
6 7 8 9 10 11 12
13 14 15 16 17 18 19
20 21 22 23 24 25 26
27 28 29 30
글 보관함