poj 2777Count Color

http://poj.org/problem?id=2777

注意：a可能比b大

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define maxn 1000010
 using namespace std;

 int  T,L,O,a,b,c;
 char ch;
 struct node
 {
     int l,r,co;
 }tree[*maxn];
 bool tab[];

 void build(int l,int r,int i)
 {
     tree[i].co=;
     tree[i].l=l;tree[i].r=r;
     if(l<r)
     {
         int mid=(l+r)/;
         build(l,mid,i+i);
         build(mid+,r,i+i+);
     }
 }

 void update(int i)
 {
     if(tree[i].co>)
     {
         tree[i+i].co=tree[i+i+].co=tree[i].co;
     }
     tree[i].co=-;
 }

 void insert1(int i,int l,int r,int co)
 {
     if(tree[i].l>=l&&tree[i].r<=r)
     {
         tree[i].co=co;
         return ;
     }
     if(tree[i].l>r||tree[i].r<l) return ;
     if(tree[i].l<tree[i].r)
     {
         int mid=(tree[i].l+tree[i].r)/;
         update(i);
         if(r<=mid) insert1(i+i,l,r,co);
         else if(l>mid) insert1(i+i+,l,r,co);
         else
         {
             insert1(i+i,l,mid,co);
             insert1(i+i+,mid+,r,co);
         }
     }
 }

 void serch(int l,int r,int i)
 {
     if(tree[i].co>)
     {
         tab[tree[i].co]=true;
         return ;
     }
     if(tree[i].l<tree[i].r)
     {
         int mid=(tree[i].l+tree[i].r)/;
         if(r<=mid) serch(l,r,i+i);
         else if(l>mid) serch(l,r,i+i+);
         else
         {
             serch(l,mid,i+i);
             serch(mid+,r,i+i+);
         }
     }
 }
 int main()
 {
     scanf("%d%d%d",&L,&T,&O);
     getchar();
     build(,L,);
     while(O--)
     {
         scanf("%c",&ch);
         if(ch=='C')
         {
             scanf("%d%d%d",&a,&b,&c);
             if(a>b)
                 swap(a,b);
             insert1(,a,b,c);
         }
         else if(ch=='P')
         {
             scanf("%d%d",&a,&b);
             memset(tab,false,sizeof(tab));
             if(a>b) swap(a,b);
             serch(a,b,);
             int num=;
             for(int i=; i<=T; i++)
             {
                 if(tab[i]) num++;
             }
             printf("%d\n",num);
         }
         getchar();
     }
     return ;
 }