zoj 1610（线段树染色问题）

题意：染色后求一段区间上各个颜色的段数

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
struct node
{
    int l,r,w,f;
}tree[8005*4+5];
int a,b,x,num[8005],last;
void build(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;
    tree[k].w=-1;
    tree[k].f=-1;
    if(l==r)
        return ;
    int mid=(l+r)/2;
    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
}
void down(int k)
{
    tree[k*2].w=tree[k].f;
    tree[k*2+1].w=tree[k].f;
    tree[k*2].f=tree[k].f;
    tree[k*2+1].f=tree[k].f;
    tree[k].f=-1;
}
void change(int k)
{
    if(tree[k].l>=a&&tree[k].r<=b)
    {
        tree[k].w=x;
        tree[k].f=x;
        return ;
    }
    if(tree[k].f!=-1) down(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(mid>=a)change(k*2);
    if(mid<b)change(k*2+1);
    if(tree[k*2].w==tree[k*2+1].w)
        tree[k].w=tree[k*2].w;
    else
        tree[k].w=-1;
}
void query(int k)
{
    if(tree[k].w!=-1||tree[k].l==tree[k].r)
    {
        if(tree[k].w!=last)
        {
            num[tree[k].w]++;
        }
        last=tree[k].w;
        return ;
    }
    if(tree[k].f!=-1)down(k);
    int mid=(tree[k].r+tree[k].l)/2;
    query(k*2);
    query(k*2+1);
}
int main()
{
    int n;
    while(cin>>n)
    {
        last=-1;
        build(1,8000,1);
        memset(num,0,sizeof(num));
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&a,&b,&x);
            a++;
            change(1);
        }
        query(1);
        for(int i=0;i<=8000;i++)
        {
            if(num[i]!=0)
                printf("%d %d\n",i,num[i]);
        }
        puts("");
    }
}