zoj 1610 Count the Colors(线段树染色）

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1610

思路：线段树区间染色，线段树查询时用color记录左边的颜色，如果和左边颜色一样那么就不属于新的一段，注意最后一个color的记录，要单独记录到ans中。数据读入中xa不代表区间，而是一个点，这样xa,xb之间的区间其实就是区间[xa+1，xb]。

#include<cstdio>
#include<cstring>
using namespace std;
const int N=8005;
struct node
{
    int l,r,f,c;
}d[N<<2];
int color,ans[N],cnt;
int max(int a,int b)
{
    if(a>b)
        return a;
    return b;
}
void pushdown(int rt)
{
    if(d[rt].l==d[rt].r)
        return ;
    d[rt].f=1;
    d[rt<<1].f=d[rt<<1|1].f=0;
    d[rt<<1].c=d[rt<<1|1].c=d[rt].c;
}
void pushup(int rt)
{
    if(d[rt].l==d[rt].r)
        return ;
    if(d[rt<<1].f==0&&d[rt<<1|1].f==0&&d[rt<<1].c==d[rt<<1|1].c)
    {
        d[rt].f=0;
        d[rt].c=d[rt<<1].c;
    }
}
void build(int rt,int L,int R)
{
    d[rt].l=L;
    d[rt].r=R;
    d[rt].f=0;
    d[rt].c=-1;
    if(L==R)
        return ;
    int mid=(L+R)>>1;
    build(rt<<1,L,mid);
    build(rt<<1|1,mid+1,R);
}
void updata(int rt,int L,int R,int col)
{
    if(d[rt].l==L&&d[rt].r==R)
    {
        d[rt].f=0;
        d[rt].c=col;
        return ;
    }
    if(d[rt].f==0)
        pushdown(rt);
    int mid=(d[rt].l+d[rt].r)>>1;
    if(R<=mid)
        updata(rt<<1,L,R,col);
    else if(L>mid)
        updata(rt<<1|1,L,R,col);
    else
    {
        updata(rt<<1,L,mid,col);
        updata(rt<<1|1,mid+1,R,col);
    }
    pushup(rt);
}
void query(int rt,int L,int R)
{
    if(d[rt].f==0)
    {
        if(d[rt].l==L&&d[rt].r==R)
        {
            if(d[rt].c!=color)
            {
                if(color>=0)
                    ans[color]++;
                color=d[rt].c;
                if(color>cnt)
                    cnt=color;
            }
            return ;
        }
        else
            pushdown(rt);
    }
    int mid=(d[rt].l+d[rt].r)>>1;
    if(R<=mid)
        query(rt<<1,L,R);
    else if(L>mid)
        query(rt<<1|1,L,R);
    else
    {
        query(rt<<1,L,mid);
        query(rt<<1|1,mid+1,R);
    }
}
int main()
{
    int n,a,b,c,i,mr;
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(scanf("%d",&n)!=EOF)
    {
        build(1,0,8000);
        mr=0;
        while(n--)
        {
            scanf("%d%d%d",&a,&b,&c);
            mr=max(mr,a);
            mr=max(mr,b);
            updata(1,a+1,b,c);
        }
        memset(ans,0,sizeof(ans));
        color=-1;cnt=0;
        query(1,1,mr);
        if(color>=0)
            ans[color]++;
        for(i=0;i<=cnt;i++)
            if(ans[i])
                printf("%d %d\n",i,ans[i]);
        printf("\n");
    }
    return 0;
}