ZOJ 1610 Count the Colors(区间染色)

题目大意：多组数据，每组给一个n(1=<n<=8000),下面有n行，每行有l,r,color(1=<color<=8000)，表示将l~r颜色变为color,最后求各种颜色（1~8000）所占区段数，如果所占区段为0则不用输出。

解题思路：基本还是跟区间染色问题差不多，但要注意一些东西

①这里l,r指的是端点。比如

3

1 4 1

1 2 2

3 4 3

这组数据最后输出结果为:

1 1

2 1

3 1

原因是还有[2,3]这一段颜色为1，处理办法是将更新函数写成（1，l+1，r，color）也就是左闭右开的形式，这样就可以使得中间的那块区间不背忽略。

②题目都没给总区间长度，所以无论查询还是建树，我们都默认用0~8000。

③关于查询各颜色所占区段数的问题，不可能直接通过线段树找到个颜色区段数，因为还要考虑两个子树是相邻的情况。所以可以通过使用一个vis[i]数组，记录各区间的颜色，最后只要简单处理一下就可以得到结果了。

#include<cstring>
#include<stdio.h>
#include<iostream>
#define LC(a) ((a<<1))
#define RC(a) ((a<<1)+1)
#define MID(a,b) ((a+b)>>1) 
using namespace std;
typedef long long ll;
const int N=8e3*4;

struct node{
    ll l,r;
    ll color;//颜色为-2表示混合色 
}tree[N];

ll ans[N];
ll vis[N]; 
//下推 
void pushdown(ll p){
    tree[RC(p)].color=tree[LC(p)].color=tree[p].color;
}
//上推 
void pushup(ll p){
    tree[p].color=(tree[LC(p)].color==tree[RC(p)].color?tree[LC(p)].color:-2);
}

void build(ll p,ll l,ll r){
    tree[p].l=l;
    tree[p].r=r;
    tree[p].color=-1;//初始化颜色因题意而定 
    if(l==r){
        return;
    }
    build(LC(p),l,MID(l,r));
    build(RC(p),MID(l,r)+1,r);
}

void update(ll p,ll l,ll r,ll color){
    if(r<tree[p].l||l>tree[p].r)
        return;
    if(tree[p].color==color)
        return;
    if(l<=tree[p].l&&r>=tree[p].r){
        tree[p].color=color;        
        return;
    }
    //**释放lazy标记 
    if(tree[p].color!=-2){
        pushdown(p);
    }
    update(LC(p),l,r,color);
    update(RC(p),l,r,color);
    pushup(p);
}

void query(ll p,ll l,ll r){
    if(r<tree[p].l||l>tree[p].r)
        return;
    
    //纯色，不用再找下去 
    if(tree[p].color!=-2){
        //**使用vis数组记录区间颜色 
        for(int i=tree[p].l;i<=tree[p].r;i++){
            vis[i]=tree[p].color;
        }
        return;
    }        
    query(LC(p),l,r);
    query(RC(p),l,r);
}

int main(){
    ios::sync_with_stdio(false);
    ll n;
    //注意是从0~n 
    while(cin>>n){
        build(1,0,8000);
        ll maxc=0;
        for(int i=1;i<=n;i++){
            ll l,r,color;
            cin>>l>>r>>color;
            maxc=max(maxc,color);
            update(1,l+1,r,color);
        }
        memset(ans,0,sizeof(ans));
        memset(vis,-1,sizeof(vis));
        query(1,1,8000);
        //处理vis数组，统计各区段数 
        for(int i=1;i<=8000;i++){
            if(vis[i]!=vis[i-1]&&vis[i]!=-1)
                ans[vis[i]]++;
        }
        for(int i=0;i<=maxc;i++){
            if(ans[i])
                cout<<i<<" "<<ans[i]<<endl;
        }
        cout<<endl;
    }
}