PS

BOJ 14245 : XOR

lickelon 2024. 11. 18. 22:49
  • 문제 링크 : boj.kr/14245
  • 난이도 : P4
  • 태그 : 느리게 갱신되는 세그먼트 트리

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

vector<ll> lazy, tree;

void push(int node, int s, int e, ll value) {
    if(value != 0) {
        tree[node] ^= value;
        if(s != e) {
            lazy[node*2] ^= value;
            lazy[node*2+1] ^= value;
        }
        lazy[node] = 0;
    }
}

void update(int node, int s, int e, int l, int r, ll value) {
    push(node, s, e, lazy[node]);
    if(e < l || r < s) return;
    if(l <= s && e <= r) {
        push(node, s, e, value);
        return;
    }
    update(node*2, s, (s+e)/2, l, r, value);
    update(node*2+1, (s+e)/2+1, e, l, r, value);
    tree[node] = tree[node*2] ^ tree[node*2+1];
}

ll query(int node, int s, int e, int l, int r) {
    push(node, s, e, lazy[node]);
    if(e < l || r < s) return 0;
    if(l <= s && e <= r) return tree[node];

    ll lq = query(node*2, s, (s+e)/2, l, r);
    ll rq = query(node*2+1, (s+e)/2+1, e, l, r);
    return lq ^ rq;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    int n;
    cin >> n;
    lazy.resize(n*4, 0);
    tree.resize(n*4, 0);

    for(int i = 1; i <= n; i++) {
        ll input;
        cin >> input;
        update(1, 1, n, i, i, input);
    }

    int m;
    cin >> m;
    for(int i = 0; i < m; i++) {
        int q;
        cin >> q;
        if(q == 1) {
            int a, b, c;
            cin >> a >> b >> c;
            update(1, 1, n, a+1, b+1, c);
        }
        if(q == 2) {
            int a;
            cin >> a;
            cout << query(1, 1, n, a+1, a+1) << "\n";
        }
    }

    return 0;
}

풀이

느리게 갱신되는 세그먼트 트리를 이용해서 XOR 값을 구해주면 된다.

728x90

'PS' 카테고리의 다른 글

BOJ 1395 : 스위치  (0) 2024.11.20
BOJ 15967 : 에바쿰  (0) 2024.11.19
BOJ 16975 : 수열과 쿼리 21  (0) 2024.11.17
BOJ 14572 : 스터디 그룹  (0) 2024.11.16
BOJ 1420 : 학교 가지마!  (5) 2024.11.15