포스트

집합(11723)

집합(11723)

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import java.io.*;
import java.util.*;

public class Main {
  
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    StringBuilder sb = new StringBuilder();
    
    int n = Integer.parseInt(br.readLine());
    
    Set<Integer> res = new HashSet<>();
    
    StringTokenizer st;
    
    for(int i = 1; i <= n; i++) {
      st = new StringTokenizer(br.readLine());

      String action = st.nextToken();
      int num = 0;
      if(!(action.equals("all") || action.equals("empty"))) num = Integer.parseInt(st.nextToken());   		
      
      switch(action) {
        case "add" : res.add(num); break;
        case "remove" : res.remove(num); break;
        case "check" : sb.append(res.contains(num)? "1\n" : "0\n"); break;
        case "toggle" : 
          if(res.contains(num)) res.remove(num);
          else res.add(num);
          break;
        case "all" : 
          for(int j = 1; j <= 20; j++) {
            res.add(j);
          }
          break;
        case "empty" : res.clear(); break;
      }
    }
    
    bw.write(sb.toString());
    bw.flush();
    bw.close();
    br.close();
  }
}

풀이과정

  1. 수가 추가 삭제가 되기 때문에 고정된 크기를 요하는 배열보다는 Set을 활용하여 풀기로 했다.
1
2
3
- 성공
- 메모리 : 313264 KB
- 시간 : 1164 ms

보안점

  1. 문제를 보면 증가 삭제되는 수의 범위가 20까지인걸 알 수 있다.
  2. 다른 사람의 문제를 참고해보면 boolean[21]를 사용하여 true, false를 이용하여 푼 것을 보았다.
  3. 범위가 주어져있고, 값의 유무만을 판단하는 경우 boolean[]을 이용하는 것도 좋은 접근인 것 같다.