ZOJ 1610 Count the Colors （线段树）

题目链接：点击链接

题意：

在x坐标轴上画线，每条线给出起点，终点，颜色编号，要求输出画了n条线后，每种颜色的线段有多少条

思路：

注意这里是[start, end]区间内为一种颜色，可以认为不包含起点终点，和以往那些包含起点终点的题目还是有区别的。处理的方法是起点加一。

• 建树的时候，当颜色把当前区间完全覆盖时，我们就不往下打。

• 查询的时候，先push down，如果当前区间在目标区间内时，遇到标记直接统计，没遇到标记就往下，直到当前区间缩为一个点。

• 如何统计每种颜色的线段的数目呢？线段树查询是先左后右的，所以我们先记录某种颜色线段出现的起点和终点，下一次遇到相同颜色的时候，我们需要判断这两条相同颜色的线段是否是相邻的，不相邻那么这个颜色的线段数就加1。

简而言之就是，利用点对应的区间进行相互判断，进行合并。

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <map>

using namespace std;

const int maxn = 8000 + 10;

struct Node
{
    int le, ri, val;
    Node(){}
    Node(int le, int ri):le(le), ri(ri){}
}A[maxn];

int color[maxn<<2];
int vis[maxn]; 
vector<Node> v[maxn];

void init(){
    memset(color, -1, sizeof(color));
    memset(vis, 0, sizeof(vis));
    for(int i=0; i<maxn; ++i) v[i].clear();
}

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

void Set(int rt, int a, int b, int l, int r, int x){
    if(a<=l && r<=b){
        color[rt] = x;
    }
    else{
        pushdown(rt);
        int mid = (l+r)>>1;
        if(a<=mid) Set(rt<<1, a, b, l, mid, x);
        if(mid<b) Set(rt<<1|1, a, b, mid+1, r, x);
    }
}

void query(int rt, int l, int r){
    if(color[rt]==-1){
        if(l==r) return;
        int mid = (l+r)>>1;
        query(rt<<1, l, mid);
        query(rt<<1|1, mid+1, r);
    }
    else{
        if(v[color[rt]].size() == 0){
            ++vis[color[rt]];
            v[color[rt]].push_back(Node(l, r));
        } 
        else{
            int tl = v[color[rt]][0].le;
            int tr = v[color[rt]][0].ri;
            if(tr+1 < l){
                ++vis[color[rt]];
            } 
            v[color[rt]][0].le = l;
            v[color[rt]][0].ri = r;
        }
    }
}


int main(){
    int n, cnt;
    while(scanf("%d", &n)!=EOF){ 
        cnt = -1;
        init();
        for(int i=0; i<n; ++i){
            scanf("%d%d%d", &A[i].le, &A[i].ri, &A[i].val);
            cnt = max(cnt, A[i].ri);
        }

        for(int i=0; i<n; ++i){
            if(A[i].ri == A[i].le) continue;
            Set(1, A[i].le+1, A[i].ri, 1, cnt, A[i].val);
        }
        query(1, 1, cnt);
        for(int i=0; i<maxn; ++i){
            if(vis[i]!=0){
                printf("%d %d\n", i, vis[i]);
            }
        }
        printf("\n");
    }
    return 0;
}