Count the Colors 线段树

题目

参考博客地址

题意：

　　n范围[1,8000] ，  li 和 ri 的范围[0,8000]。  n个操作，每个操作是把 [li , ri]内的点修改成一个颜色c。 n个操作过后，按颜色从小到大 输出每种颜色分别有几块。

 #include<iostream>
 #include<cstdio>
 #include <cctype>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #include<string>
 #include<cmath>
 #include<set>
 #include<vector>
 #include<stack>
 #include<queue>
 #include<map>
 using namespace std;
 #define ll long long
 #define mem(a,x) memset(a,x,sizeof(a))
 #define se second
 #define fi first
 const ll mod=1e9+;
 const int INF= 0x3f3f3f3f;
 const int N=1e5+;

 int add[N<<];
 int flag; //flag判断相邻两个区间的颜色是否相同
 int ans[N<<];

 void push_down(int rt)
 {
     if(add[rt]>)
     {
         add[rt<<]= add[rt];
         add[rt<<|]= add[rt];
         add[rt]=-;
     }
 }
 void Built(int l,int r,int rt)
 {
     add[rt]=;
     if(l==r) return;
     int m=l+r>>;
     Built(l,m,rt<<);
     Built(m+,r,rt<<|);
 }

 void update(int x,int y,int c,int l,int r,int rt)
 {
     if(x<=l && r<=y)
     {
         add[rt]=c;
         return;
     }
     push_down(rt);
     int m=l+r>>;
     if(x<=m) update(x,y,c,l,m,rt<<); //当寻找一个区间的时候，路径上的值全改成c
     if(m<y) update(x,y,c,m+,r,rt<<|);//当寻找一个区间的时候，路径上的值全改成c
     add[rt]=-; //寻找到了之后，把回头的路径全部改成-1，说明如果顺着这些点下来，一定能找到区间
 }

 void query(int l,int r,int rt)
 {
     if(flag==add[rt]) return;
     if(add[rt]==) //一次也没有被涂过
     {
         flag=; return;
     }
     if(add[rt]>)
     {
         if(flag!=add[rt]) //不是同一块海报
         {
             flag=add[rt];
             ans[add[rt]]++;
         }
         return;
     }

     //接下来是如果add[rt]== -1 ，表示顺着这个点 一定能找到区间
     if(l==r)
         return;
     int m=l+r>>;
     query(l,m,rt<<);
     query(m+,r,rt<<|);
 }
 int main()
 {
     int x,y,z,n,m;
     while(scanf("%d",&n)==)
     {
         mem(add,); //0表示该点没有涂色。
         mem(ans,);
         Built(,,);  

         for(int i=;i<=n;i++)
         {
             scanf("%d%d%d",&x,&y,&z);
             update(x, y-, z+, ,,);
         }
         query(,,);

         for(int i=;i<=;i++)
         {
             if(ans[i]) printf("%d %d\n",i-,ans[i]);
         }
         cout<<endl;
     }
 }