[B15787] - 기차가 어둠을 헤치고

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

비트마스크로 구현하면 되며 vector의 크기를 결정을 잘 해야 한다.

vector<int>d에는 2^1부터 2^20승이 더한 값이 들어갈 수 있으므로 2^21의 크기로 잡아주었다.

| 는 승차, &~는 하차를 의미한다.

#include <iostream>
#include <vector>
using namespace std;
int main()
{
    int n, m;
    cin >> n >> m;
 
    vector<int>d((1 << 21));
    vector<int>train(n+1);
    int ans = 0;
    for (int i = 0; i < m; i++)
    {
        int u;
        int a, b;//a번째 기차에서 b번째 좌석
        cin >> u;
        if (u == 1)
        {
            cin >> a >> b;
            train[a] = (train[a] | (1 << b));
        }
        else if (u == 2)
        {
            cin >> a >> b;
            train[a] = (train[a] & ~(1 << b));
        }
        else if(u==3)
        {
            cin >> a;
            if (train[a] & (1 << 20))
                train[a] = train[a] & ~(1 << 20);
            train[a]=train[a] << 1;
        }
        else if (u == 4)
        {
            cin >> a;
            train[a] = train[a] >> 1;
            train[a] = (train[a] & ~(1 << 0));
        }
    }
    for (int i = 1; i < n+1; i++)
    {
        d[train[i]]++;
    }
    int len = (1 << 21);
    for (int i = 0; i < len; i++)
    {
        if (d[i])
        {
            ans++;
        }
    }
    cout << ans << '\n';
}