zoj1610 count the colors(线段树）

题目大意:n次操作，每次对一段区间进行染色。问最后有每种颜色有多少段。

解题思路:对区间颜色进行维护。注意，区间应该是从1到8000的（每次输入的区间是左开右闭区间，是个坑），颜色为0到8000。n次更新后，查询每个点对应的颜色，若跟前一个点颜色不同，则进行计数，最后输出即可。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 1e4+5;
int segment[4*maxn],cnt[maxn];
int n;

void pushdown(int i)
{
    if(segment[i])
    {
        segment[i<<1] = segment[i];
        segment[i<<1|1] = segment[i];
        segment[i] = 0;
    }
}

void build(int i,int l,int r)
{
    if(l == r)
    {
        segment[i] = 0;
        return;
    }

    int mid = (l + r) >> 1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
}

void update(int i,int val,int ql,int qr,int l,int r)
{
    if(ql > r || qr < l)    return;
    if(ql <= l && qr >= r) 
    {
        segment[i] = val;
        return;
    }
    pushdown(i);

    int mid = (l + r) >> 1;
    update(i<<1,val,ql,qr,l,mid);
    update(i<<1|1,val,ql,qr,mid+1,r);
}

int query(int i,int x,int l,int r)
{
    if(l == r)  return segment[i];
    pushdown(i);

    int mid = (l + r) >> 1;
    if(x <= mid)
        return query(i<<1,x,l,mid);
    else 
        return query(i<<1|1,x,mid+1,r);
}

void solve()
{
    int last = -1;
    for(int i = 1;i <= 8000;++i)
    {
        int temp = query(1,i,1,8000);
        // cout<<temp<<' ';
        if(last != temp)    cnt[temp]++;
        last = temp;
    }
    for(int i = 1;i <= 8001;++i)
        if(cnt[i]) cout<< (i - 1) <<' '<<cnt[i]<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);cout.tie(NULL);

    while (cin>>n)
    {
        build(1,1,8000);
        memset(cnt,0,sizeof(cnt));
        
        int a,b,c;
        for(int i = 1;i <= n;++i)
        {
            cin>>a>>b>>c;
            update(1,c + 1,a + 1,b,1,8000);
        }
        solve();
        cout<<endl;
    }
    
    return 0;
}