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*;

 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:-);
 }

 void build(ll p,ll l,ll r){
     tree[p].l=l;
     tree[p].r=r;
     tree[p].color=-;//初始化颜色因题意而定
     if(l==r){
         return;
     }
     build(LC(p),l,MID(l,r));
     build(RC(p),MID(l,r)+,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!=-){
         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!=-){
         //**使用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(,0,);
         ll maxc=;
         for(int i=;i<=n;i++){
             ll l,r,color;
             cin>>l>>r>>color;
             maxc=max(maxc,color);
             update(,l+,r,color);
         }
         memset(ans,,sizeof(ans));
         memset(vis,-,sizeof(vis));
         query(,,);
         //处理vis数组，统计各区段数
         for(int i=;i<=;i++){
             if(vis[i]!=vis[i-]&&vis[i]!=-)
                 ans[vis[i]]++;
         }
         for(int i=;i<=maxc;i++){
             if(ans[i])
                 cout<<i<<" "<<ans[i]<<endl;
         }
         cout<<endl;
     }
 }