티스토리 뷰

알고리즘 관련/BOJ

BOJ)1395 스위치

JASON 자손9319 2017. 1. 13. 16:59

문제: icpc.me/1395


1~N번까지 스위치가 존재할 때 두가지 쿼리에 따른 일을 한다.


1. S~T까지 스위치를 반전 시키는 것

2. S~T까지 켜져있는 스위치의 개수를 출력하는 것


우리는 세그먼트 트리의 구간합을 통하여 켜져있는 스위치의 개수를 쉽게 구할 수 있다.


하지만 1번 쿼리의 경우 구간에 대한 갱신이 일어나기 때문에 우리는 lazy propagation을 사용하여 업데이트를 시켜줘야만 한다.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include <cstdio>
#include <algorithm>
#define MAX_N 100000
using namespace std;
int n, m, seg[* MAX_N], lazy[* MAX_N], o, s, t;
void u_lazy(int node, int x, int y) {
    if (!lazy[node])
        return;
    seg[node] = (y - x + 1- seg[node];
    if (x != y) {
        lazy[node * 2] ^= 1;
        lazy[node * + 1] ^= 1;
    }
    lazy[node] = 0;
}
int update(int lo, int hi, int node, int x, int y) {
    u_lazy(node, x, y);
    if (y < lo || hi < x)
        return seg[node];
    if (lo <= x&&<= hi) {
        lazy[node] ^= 1;
        u_lazy(node, x, y);
        return seg[node];
    }
    int mid = (x + y) >> 1;
    return seg[node] = update(lo, hi, node * 2, x, mid) + update(lo, hi, node * + 1, mid + 1, y);
}
int query(int lo, int hi, int node, int x, int y) {
    u_lazy(node, x, y);
    if (y < lo || hi < x)
        return 0;
    if (lo <= x&&<= hi)
        return seg[node];
    int mid = (x + y) >> 1;
    return query(lo, hi, node * 2, x, mid) + query(lo, hi, node * + 1, mid + 1, y);
}
int main() {
    scanf("%d%d"&n, &m);
    for (int i = 0; i < m; i++) {
        scanf("%d%d%d"&o, &s, &t);
        if (o)
            printf("%d\n", query(s, t, 11, n));
        else
            update(s, t, 11, n);
    }
    return 0;
}
cs


'알고리즘 관련 > BOJ' 카테고리의 다른 글

BOJ)3392 화성 지도  (1) 2017.01.13
BOJ)2336 굉장한 학생  (0) 2017.01.13
BOJ)1395 스위치  (0) 2017.01.13
BOJ)10999 구간 합 구하기2  (1) 2017.01.13
BOJ)11505 구간 곱 구하기  (2) 2017.01.12
BOJ)5012 불만 정렬  (2) 2017.01.11
댓글
댓글쓰기 폼