ZOJ - 1610 Count the Colors(线段树)

题目大意：给你N个操作，操作如下
l r c,将[l,r]涂成c这种颜色
问最后有颜色的占的长度是多大

解题思路：线段树的区间修改，最后再询问一下就可以了
但这题有个坑点，就是他要求染的是一个区间
假设给出如下数据
1 4 1
1 2 2
3 4 3
如果直接套用他所给的数的话，得到的答案是只有两种颜色的，但实际上有三种颜色，[2,3]是颜色1的，所以在Modify的时候要特殊处理一下，把l+1就好了，因为他要问区域指的的长度，比如[3,4]的话，其实只占1个单位长度而已

#include <cstdio>
#include <cstring>

const int N = 8010 << 2;
int n;
int color[N], segment[N], ans[N];

void PushDown(int u) {
    if (color[u] != -1) {
        color[u << 1] = color[u << 1 | 1] = color[u];
        color[u] = -1;
    }
}

void modify(int u, int l, int r, int L, int R, int c) {

    if (L <= l && r <= R) {
        color[u] = c;
        return ;
    }
    if (color[u] == c) return ;
    PushDown(u);
    int mid = (l + r) >> 1;
    if (L <= mid) modify(u << 1, l, mid, L, R, c);
    if (R > mid) modify(u << 1 | 1, mid + 1, r, L, R, c);
}

void query(int u, int l, int r) {
    if (color[u] != -1) {
        for (int i = l; i <= r; i++)
            segment[i] = color[u];
        return ;
    }
    if (l == r) return ;
    int mid = (l + r) >> 1;
    query(u << 1, l, mid);
    query(u << 1 | 1, mid + 1, r);
}

void solve() {

    memset(color, -1, sizeof(color));
    memset(segment, -1, sizeof(segment));
    memset(ans, 0, sizeof(ans));

    int a, b, c;
    while (n--) {
        scanf("%d%d%d", &a, &b, &c);
        if (a >= b) continue;
        modify(1, 1, 8000, a + 1, b, c);
    }
    query(1, 1, 8000);
    int i = 1;
    while (i <= 8000) {
        int c = segment[i], j = i + 1;
        if (c == -1) {
            i++; 
            continue;
        }
        while (segment[j] == c && j <= 8000) j++;
        ans[c]++;
        i = j;
    }
    for (int i = 0; i <= 8000; i++)
        if (ans[i]) printf("%d %d\n", i, ans[i]);
    printf("\n");
}

int main() {
    while (scanf("%d", &n) != EOF) solve();

    return 0;
}