POJ 2777 Count Color 线段树

线段树的模板题。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <algorithm>

using namespace std;

struct N
{
    int l,r,col;
    bool MarkVisit;
} st[400100];

void init(int l,int r,int site)
{
    st[site].l = l;
    st[site].r = r;
    st[site].col = 1;
    st[site].MarkVisit = true;//标记为单色

    if(l == r)
        return ;
    int mid = (l+r)>>1;

    init(l,mid,site<<1);
    init(mid+1,r,site<<1|1);
}

void updata(int l,int r,int col,int site)
{
    if(l == st[site].l && r == st[site].r)
    {
        st[site].col = col;
        st[site].MarkVisit = true;
        return ;
    }

    if(st[site].MarkVisit && st[site].col == col)
        return ;

    int mid = (st[site].l + st[site].r)>>1;

    if(st[site].MarkVisit)
    {
        updata(st[site].l,mid,st[site].col,site<<1);
        updata(mid+1,st[site].r,st[site].col,site<<1|1);
    }

    st[site].MarkVisit = false;

    if(r <= mid)
    {
        updata(l,r,col,site<<1);
    }
    else if(mid < l)
    {
        updata(l,r,col,site<<1|1);
    }
    else
    {
        updata(l,mid,col,site<<1);
        updata(mid+1,r,col,site<<1|1);
    }
}

bool MarkColor[40];

int query(int l,int r,int site)
{
    if(st[site].l <= l && r <= st[site].r && st[site].MarkVisit == true)
    {
        if(MarkColor[st[site].col] == false)
        {
            MarkColor[st[site].col] = true;
            return 1;
        }
        return 0;
    }

    int mid = (st[site].l + st[site].r)>>1;

    if(r <= mid)
    {
        return query(l,r,site<<1);
    }
    if(mid < l)
    {
        return query(l,r,site<<1|1);
    }
    return query(l,mid,site<<1) + query(mid+1,r,site<<1|1);

}

int main()
{
    int len,t,o;

    int l,r,col;

    char op;

    while(scanf("%d %d %d",&len,&t,&o) != EOF)
    {
        init(1,len,1);//初始化
        while(o--)
        {
            scanf("%*c%c",&op);
            if(op == 'C')
            {
                scanf("%d %d %d",&l,&r,&col);
                updata(l,r,col,1);
            }
            else
            {
                scanf("%d %d",&l,&r);

                memset(MarkColor,false,(t+2)*sizeof(bool));

                printf("%d\n",query(l,r,1));
            }
        }
    }
    return 0;
}