ZOJ1610 Count the Colors

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1610

貌似线段树的区间染色是一个非常常见的题型吧，但是不得不说。。我昨天刚刚碰到，而且统计每一种颜色占了多少个不连续区间的时候着实不会写，表示前面把树建出来和更新的步骤都是纯我自己写的，就是查找颜色个数那一步。。。不得不参考了，最后弄明白了。

如果一个区间没有染色或者有多种颜色，都会有标记，如果有了标记，就可以不用考虑这个结点了，直接找它的两个儿子就行了。如果证实某一个区间被染过色而且里面仅仅有一个颜色，那么就可以与其相邻子树比较然后再更新结果值（要求的是每种颜色占的不连续区间数量，相邻子树代表的是连续区间，所以就得这么比较），然后直接退出这个状态（都说了里面只有一个颜色，就没有找它儿子的必要了）。。

#include <cstdio>#include <cstring>#define lson l, m, rt << 1#define rson m, r, rt << 1 | 1const int maxn = 8111;int col[maxn << 2], tmp, num[maxn];void PushDown(int rt){    if (col[rt] >= 0){        col[rt << 1] = col[rt];        col[rt << 1 | 1] = col[rt];        col[rt] = -2;    }}void build(int l, int r, int rt){    col[rt] = -1;    if (l == r - 1) return;    int m = (l + r) >> 1;    build(lson);    build(rson);}void update(int ll, int rr, int c, int l, int r, int rt){    if (col[rt] == c) return;    if (ll <= l && rr >= r){        col[rt] = c;        return;    }    PushDown(rt);    int m = (l + r) >> 1;    if (ll < m) update(ll, rr, c, lson);    if (rr > m) update(ll, rr, c, rson);}void query(int l, int r, int rt){    if (col[rt] != -1 && col[rt] != -2){        if (col[rt] != tmp) num[col[rt]] += 1;        tmp = col[rt];        return;    }    if (l == r - 1){        tmp = -1; return;    }    int m = (l + r) >> 1;    query(lson); query(rson);}int main(){    //freopen("in.txt", "r", stdin);    int n;    while(~scanf("%d", &n)){        build(0, 8000, 1);        memset(num, 0, sizeof(num));        while(n--){            int x, y, z;            scanf("%d%d%d", &x, &y, &z);            update(x, y, z, 0, 8000, 1);        }        tmp = -1;        query(0, 8000, 1);        for (int i = 0; i <= 8000; i++)            if (num[i]) printf("%d %d\n", i, num[i]);        printf("\n");    }    return 0;}