뚱땅뚱땅

[문제] 백준 11723번 집합 본문

알고리즘/백준

[문제] 백준 11723번 집합

양순이 2021. 7. 14. 14:28
728x90

https://www.acmicpc.net/problem/11723

 

11723번: 집합

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다. 둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

www.acmicpc.net

 

string도 switch문으로 비교 가능하다는 것을 알게 됐다.

또한, HashSet을 적절히 이용하는 방법을 연습할 수 있었다.

public class Main {	
	public static void main(String[] args) throws NumberFormatException, IOException{
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		
		int M = Integer.parseInt(in.readLine()); // 연산의 개수
		HashSet<Integer> set = new HashSet<>();
		StringBuilder sb = new StringBuilder();
		
		for(int i=0;i<M;i++) {
			StringTokenizer st = new StringTokenizer(in.readLine()," ");
			String inst = st.nextToken(); // 명령어
			int num=0; // 임시 초기화
			
			switch(inst) {
				case "add":
					num = Integer.parseInt(st.nextToken());
					set.add(num);
					break;
				case "remove":
					num = Integer.parseInt(st.nextToken());
					set.remove(num);
					break;
				case "check":
					num = Integer.parseInt(st.nextToken());
					sb.append(set.contains(num)? "1":"0").append('\n');
					break;
				case "toggle":
					num = Integer.parseInt(st.nextToken());
					if(set.contains(num)) {
						set.remove(num);
					}else {
						set.add(num);
					}
					break;
				case "all":
					set.clear();
					for(int j=1;j<=20;j++)
						set.add(j);
					break;
				case "empty":
					set.clear();
					break;
			}
		}
		System.out.print(sb);
	}
	
}
728x90
Comments