[백준 11723, c++] 집합

문제 번호

• 11723(https://www.acmicpc.net/problem/11723)

문제 및 입/출력

비어있는 공집합 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) 배열을 이용한 풀이

/* 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, true, sizeof(result));
        }
        else {
            memset(result, false, sizeof(result));
        }
    }
    return 0;
}

/* 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;
}