백준 15787번 기차가 어둠을 헤치고 은하수를 C++

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

15787번: 기차가 어둠을 헤치고 은하수를

 

비트를 이용해서 겹치지 않는 좌석의 기차만 카운팅했다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;
vector<int> v;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> m;
    v = vector<int>(n + 1, 0);
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b;
        if (a <= 2) {
            cin >> c;
            if (a == 1) {
                v[b] |= (1 << c);
            }
            else if (a == 2) {
                v[b] &= ~(1 << c);
            }
        }
        else {
            if (a == 3) {
                v[b] <<= 1;
                v[b] &= ~(1 << 21);
            }
            else if (a == 4) {
                v[b] >>= 1;
                v[b] &= ~1;
            }
        }
    }
    set<int> s;
    for (int i = 1; i < v.size(); i++) {
        s.insert(v[i]);
    }
    cout << s.size();
}