[BOJ] 11723번 집합 (C++)

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

11723번: 집합

 

#include <iostream>
#include <set>
#include <string>
using namespace std;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
    
	int m;
	cin >> m;
	
	set<int> s;
	set<int> tmp;
	for(int i = 1; i <= 20; i++) tmp.insert(i);
	string str;
	int k;
	for(int i = 0; i < m; i++){
		cin >> str;
		if(str == "add"){
			cin >> k;
			s.insert(k);
		} else if(str == "remove"){
			cin >> k;
			s.erase(k);
		} else if(str == "check"){
			cin >> k;
			if(s.find(k) == s.end()){
				cout << "0\n";
			} else{
				cout << "1\n";
			}
		} else if(str == "toggle"){
			cin >> k;
			if(s.find(k) == s.end()){
				s.insert(k);
			} else{
				s.erase(k);
			}
		} else if(str == "all"){
			s = tmp;
		} else if(str == "empty"){
			s.clear();
		}
	}
	return 0;
}

all일 때마다 for문을 돌리니까 시간 초과가 발생했다.

{1, 2, ..., 20}인 집합 tmp를 미리 만들어두고 all일 때 s에 tmp를 대입하여 시간 초과를 해결했다.