poj 2777 Count Color (线段树)

#include<iostream>
#include<math.h>
#include<cstdio>
#include<stdlib.h>
#include<cstring>
using namespace std;
 #define MAXN 100010
 /*
 update是把一个区间内颜色相同的改成相同颜色
 颜色不同为-1

 query 是查询函数，如果颜色不是-1，就把这个颜色记录为1
 最后所有在这个区间的颜色都加一起
 */
 struct node
 {
     int l,r,s;
 }t[MAXN<<2];
 int L,O,T;
 int sum[33];
 void build(int l,int r,int i)
 {
     t[i].l=l;
     t[i].r=r;
     t[i].s=1;
     if(l==r)return ;
     int mid =(l+r)>>1;
     build(l,mid,i<<1);
     build(mid+1,r,i<<1|1);//就是i*2+1
 }

 void update(int l, int r, int m, int i)
{
    if(t[i].s == m) return ;
    if(t[i].l == l && t[i].r == r)
    {
        t[i].s = m;
        return ;
    }
    if(t[i].s != -1)
    {
        t[i<<1].s = t[i<<1|1].s = t[i].s;
        t[i].s = -1;
    }
    int mid = (t[i].l+t[i].r)>>1;
    if(l > mid) update(l, r, m, i<<1|1);
    else if(r <= mid) update(l, r, m, i<<1);
    else
    {
        update(l, mid, m, i<<1);
        update(mid+1, r, m, i<<1|1);
    }
}


 void query(int l, int r, int i)
{
    if(t[i].s != -1)//如果纯色把该颜色标记
    {
        sum[t[i].s] = 1;
        return ;
    }
    else//否则继续查询子节点
    {
        int mid = (t[i].l+t[i].r)>>1;
        if(l > mid) query(l, r, i<<1|1);
        else if(r <= mid) query(l, r, i<<1);
        else
        {
            query(l, mid, i<<1);
            query(mid+1, r, i<<1|1);
        }
    }
}


int main() {
    char ch;
  int a,b,c;
int L,T,O;
    while(~scanf("%d%d%d",&L,&T,&O)) {
        build(1,L,1);
        while(O--) {
                getchar();
        scanf("%c",&ch);
        if(ch=='C')
            {scanf("%d %d %d",&a,&b,&c);
             update(a,b,c,1);}
          int ans=0;
          if(ch=='P')
          {
              memset(sum,0,sizeof(sum));
              scanf("%d%d",&a,&b);
              query(a,b,1);
              for(int i=1;i<=T;i++)
              {
                  if(sum[i])ans++;
              }
              printf("%d\n",ans);
          }
        }
    }
    return 0;
}