곧죽어도 콛잉

[비트마스킹 / BOJ 11723 / C++] 집합 본문

Coding Test/C++

[비트마스킹 / BOJ 11723 / C++] 집합

코드진행형 2023. 5. 6. 00:06

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

 

11723번: 집합

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다. 둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

www.acmicpc.net

1) 집합의 개념은 비트마스킹으로 생각하면 된다. 즉 S = 0x00000 ~ 0xFFFFF 까지 가능.

간만에 푼 진짜 쉬웠던 문제!! 크게 설명할 것 없이 비트마스킹인 점을 생각하고 문제를 접근하면 쉽다!

 

000..0000 ~ 1111..11111 까지 생각하여 그 숫자의 존재를 있다면 1, 없다면 0으로 표시하여 진행하면 된다!! 

( 16진수로 적으면, 0x00000 ~ 0xFFFFF )

 

바로 코드를 보면서 이해해보자!

 

check 같은 경우 삼항 연산자를 통해 더욱 간단하게 써줄 수도 있다.

모든 비트를 키는 경우는 그냥 0xFFFFF로 지정해주긴 했는데 좀 더 비트마스킹을 이용하고 싶다면

S = ((1 << 21) - 1) 를  이용하면 된다.

#include <bits/stdc++.h>

using namespace std;

int M, S;


int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    
    cin >> M;
    for(int i = 0 ; i < M ; i++){
        string tmp;
        int idx;
        
        cin >> tmp;
        
        if(tmp == "add"){
            cin >> idx;
            idx = idx - 1;
            S |= (1<<idx); // idx번째 bit on
        }else if(tmp == "remove"){
            cin >> idx;
            idx = idx - 1;
            S &= ~(1<<idx); // idx 번째 bit off
        }else if(tmp == "check"){
            cin >> idx;
            idx = idx - 1;
            if(S & (1<<idx)) cout << 1 << '\n';
            else cout << 0 << '\n';
        }else if(tmp == "toggle"){
            cin >> idx;
            idx = idx - 1;
            if((S & (1<<idx))){ // idx번째가 켜져있다면 끄기
                S &= ~(1<<idx);
            } else{ // 꺼져있다면 키기.
                S |= (1<<idx);
            }
        }else if(tmp=="all"){
            S = 0xFFFFF;
        }else if(tmp=="empty"){
            S = 0;
        }
    }
    
    
    
    
    
}
Comments