POJ 2777 Count color 树状数组

#include <iostream>

#include<cstdio>
#include<algorithm>


using namespace std;
const int MAX=100003;
struct node
{
    int col;
    bool book;
    int l,r;
}tre[MAX*4];
int sum;
void build(int s,int e,int rot)
{
    tre[rot].l=s;
    tre[rot].r=e;
    tre[rot].col=1;
    tre[rot].book=1;
    if(s==e)
        return;


    int mid=(s+e)/2;
    build(s,mid,rot*2);
    build(mid+1,e,rot*2+1);
}
void pushdown(int rt)
{
    tre[rt*2].col=tre[rt].col;
    tre[rt*2+1].col=tre[rt].col;
    tre[rt*2].book=1;
    tre[rt*2+1].book=1;
    tre[rt].book=0;
}
void update(int s,int e,int c,int rot)
{
    if(s<=tre[rot].l&&e>=tre[rot].r)
    {
        tre[rot].book=1;
        tre[rot].col=c;         
        return;
    }
    if(tre[rot].col==c)   //如果此时本节点区间内为一个颜色就且等于待更新颜色
        return;           //直接返回了,不用更新
    if(tre[rot].book)     //如果此时的覆盖成立，则向下更新子节点
        pushdown(rot);


    int mid=(tre[rot].l+tre[rot].r)/2;
    if(mid>=e)
        update(s,e,c,rot*2);
    else if(s>mid)
        update(s,e,c,rot*2+1);
    else
    {
        update(s,mid,c,rot*2);
        update(mid+1,e,c,rot*2+1);
    }
    tre[rot].col=tre[rot*2].col|tre[rot*2+1].col;
}
void quiry(int s,int e,int rt)
{
    if(s<=tre[rt].l&&e>=tre[rt].r)
    {
        sum|=tre[rt].col;
        return;
    }
    if(tre[rt].book)
    {
        sum|=tre[rt].col;
        return;
    }


    int mid=(tre[rt].l+tre[rt].r)/2;    //注意此处是根据根节点的mid分区间检查
    if(mid>=e)
        quiry(s,e,rt*2);
    else if(s>mid)
        quiry(s,e,rt*2+1);
    else                           //注意一旦分区间查，要把区间分开去查下去
    {
        quiry(s,mid,rt*2);
        quiry(mid+1,e,rt*2+1);
    }
}
int solve(int sum)
{
    int s=0;
    while(sum)
    {
        if(sum&1)
            s++;
        sum>>=1;
    }
    return s;
}
int main()
{
    int L,T,O;
    scanf("%d%d%d",&L,&T,&O);
    build(1,L,1);
    char o;
    while(O--)
    {
        int s,e,c;
        getchar();
            scanf("%c",&o);
            scanf("%d%d",&s,&e);
        if(s>e)
            s^=e^=s^=e;
        if(o=='C')
        {
            scanf("%d",&c);
            update(s,e,1<<(c-1),1);
        }
        else
        {
            sum=0;
            quiry(s,e,1);
            printf("%d\n",solve(sum));
        }
    }
    return 0;
}