#11723 : 집합

풀이방법

사용된 것:

HashSet

이 문제에서는 중복값을 허용하지 않는 자료구조를 사용하는 것이 편리하고

정렬은 필요 없기 때문에

해시셋을 사용했다.

 

2022.03.28

연산이 한 줄에 하나씩 주어질 때마다

StringTokenizer를 사용하여 첫 번째 단어를 가져온다.

이 첫 번째 단어를 기준으로 switch case문을 돌린다.

 

단, 시간초과에 유의해야 한다.

 

check 연산 시마다 System.out.println()을 실행하면 시간초과가 발생한다.

더 빠른 출력 방식을 사용해야 한다.

아래 코드에서는 StringBuilder를 이용하였다.

 

all 연산에 드는 시간도 줄이기 위하여,

for(int j = 1; j<21; j++) 를 매번 돌리는 대신에

1부터 20까지의 수가 저장된 해시셋을 맨 처음에 따로 미리 만들어두고

all 연산이 요청되면 이 해시셋을 통째로 가져다 붙이는 방식을 사용했다.

 

코드

Java(2022.03.28)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException{
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		HashSet<Integer> hs = new HashSet<Integer>();
		HashSet<Integer> all = new HashSet<Integer>();
		for(int i = 1; i<21; i++) all.add(i);
		
		int m = Integer.parseInt(br.readLine());
		
		StringBuilder sb = new StringBuilder();
		for(int i = 0; i<m; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			
			String command = st.nextToken();
			
			switch(command) {
			
			case "add": hs.add(Integer.parseInt(st.nextToken()));
				break;
			case "remove": hs.remove(Integer.parseInt(st.nextToken()));
				break;
			case "check": if(hs.contains(Integer.parseInt(st.nextToken()))) sb.append(1 + System.lineSeparator());
				else sb.append(0 + System.lineSeparator());
				break;
			case "toggle": int num = Integer.parseInt(st.nextToken());
				if(hs.contains(num)) hs.remove(num);
				else hs.add(num);
				break;
			case "all": hs = all;
				break;
			case "empty": hs.clear();
				break;
			}
		}
		
		System.out.print(sb);
	}

}