백준/브루트 포스

[백준 11723, c++] 집합

전두선 2019. 8. 20. 11:34
문제 번호
문제 및 입/출력

비어있는 공집합 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을 출력한다.
  • toggle x: S에 x가 있으면 x를 제거하고, 없으면 x를 추가한다. (1 ≤ x ≤ 20)
  • all: S를 {1, 2, ..., 20} 으로 바꾼다.
  • empty: S를 공집합으로 바꾼다. 

입력

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.

둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

 

출력

check 연산이 주어질때마다, 결과를 출력한다.

 

예제 입력
26
add 1
add 2
check 1
check 2
check 3
remove 2
check 1
check 2
toggle 3
check 1
check 2
check 3
check 4
all
check 10
check 20
toggle 10
remove 20
check 10
check 20
empty
check 1
toggle 1
check 1
toggle 1
check 1

 

예제 출력
1
1
0
1
0
1
0
1
0
1
1
0
0
0
1
0

 

문제 풀이

이 문제는 배열을 이용하여 풀 수도 있고, 비트연산자로도 풀 수가 있다. 

 

연산의 수의 범위가 M (1 ≤ M ≤ 3,000,000)이기에 이 문제는 제한 시간 조건을 따지는게 가장 중요한 포인트라고 생각한다. 인터넷에 자료를 찾아본 결과 내가 자주 사용했던 cin,cout은 scanf, print에 비해 느리고, scanf, printf는 gets,puts에 비해 느리다고 한다.

보통 문제들은 결과 값 하나만 출력하기에 cin,cout을 사용해도 지장이 없지만 본 문제처럼 주기적으로 출력을 요구하는 문제의 경우 cin,cout을 사용한다면 지연으로 인해 문제가 될 수 있다는 것을 항상 알고 있어야한다.

 

소스 코드

(1) 배열을 이용한 풀이

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
47
48
49
50
51
52
53
54
55
56
57
/* https://www.acmicpc.net/problem/11723 */
 
#include <iostream>
#include <string>
#include <cstring>
 
using namespace std;
 
int main(void) {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    const int MAX = 20 + 1;
    bool result[MAX] = { 0 };
 
    int m;
    cin >> m;
    for (int i = 0; i < m; i++) {
        int tmp;
        string str;
        cin >> str;
 
        if (str == "add") {
            cin >> tmp;
 
            result[tmp] = true;
        }
        else if (str == "remove") {
            cin >> tmp;
 
            result[tmp] = false;
        }
        else if (str == "check") {
            cin >> tmp;
 
            if (result[tmp])
                cout << "1\n";
            else
                cout << "0\n";
        }
        else if (str == "toggle") {
            cin >> tmp;
 
            if (result[tmp])
                result[tmp] = false;
            else
                result[tmp] = true;
        }
        else if (str == "all") {
            memset(result, truesizeof(result));
        }
        else {
            memset(result, falsesizeof(result));
        }
    }
    return 0;
}

 

(2) 비트연산자

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
47
48
49
50
51
52
53
54
/* https://www.acmicpc.net/problem/11723 */
 
#include <iostream>
#include <string>
#include <cstring>
 
using namespace std;
 
int main(void) {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    int s = 0;
 
    int m;
    cin >> m;
    for (int i = 0; i < m; i++) {
        int tmp;
        string str;
        cin >> str;
 
        if (str == "add") {
            cin >> tmp;
            s |= (1 << tmp);
        }
        else if (str == "remove") {
            cin >> tmp;
            s &= ~(1 << tmp);
        }
        else if (str == "check") {
            cin >> tmp;
 
            if (s & (1 << tmp))
                cout << "1\n";
            else
                cout << "0\n";
        }
        else if (str == "toggle") {
            cin >> tmp;
 
            if (s & (1 << tmp))
                s &= ~(1 << tmp);
            else
                s |= (1 << tmp);
        }
        else if (str == "all") {
            s = (1 << 21- 1;
        }
        else {
            s = 0;
        }
    }
    return 0;
}

 

 

'백준 > 브루트 포스' 카테고리의 다른 글

[백준 2309, c++] 일곱난쟁이  (0) 2019.08.16