Count the Colors---zoj1610线段树

题目链接

题意：

求每种颜色有几段线段；

模拟数组：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
#define N 8006
int cnt[N],a[N];
int main()
{
    int n, p, q, c;
    while(scanf("%d", &n) != EOF)
    {
        int Max = ;
        memset(a, , sizeof(a));
        memset(cnt, , sizeof(cnt));
        for(int i=; i<n; i++)
        {
            scanf("%d%d%d", &p, &q, &c);
            for(int j=p; j<q; j++)
                a[j] = c+;
            Max = max(Max, q);
        }
        for(int i=; i<Max; i++)
        {
            while(i!= && a[i]!= && a[i]==a[i-])
                i++;
            if(a[i] != )
                cnt[ a[i]- ]++;
        }
        for(int i=; i<N; i++)
        {
            if(cnt[i]!=)
                printf("%d %d\n", i, cnt[i]);
        }
        printf("\n");
    }
    return ;
}

#include<stdio.h>
#include<string.h>
#define N 8005

struct SegTree
{
    int L,R,c;
    int mid()
    {
        return (L+R)>>;
    }
}a[N*];

int cnt[N], vis[N];

void BuildSegTree(int r, int L, int R)
{
    a[r].L = L;
    a[r].R = R;
    a[r].c = -;
    if(L+==R)//一条线段的两个端点相差1，
        return ;
    BuildSegTree(r*, L, a[r].mid());
    BuildSegTree(r*+, a[r].mid(), R);//题目给的是端点值，对于线段不能用a[r].mid()+1;
}
void Update(int r, int L, int R, int c)
{
    if(a[r].c == c)return ;//如果说这条线段的颜色和要涂的相同，返回；
    if(a[r].L == L && a[r].R == R)
    {
        a[r].c = c;//指定范围内改成颜色c；
        return ;
    }
    if(a[r].c != -)
    {
        a[*r].c = a[*r+].c = a[r].c;//更新左右子树；
        a[r].c = -;
    }

    if(R<=a[r].mid())
        Update(*r, L, R, c);
    else if(L>=a[r].mid())//等于号不能省；原因是这是线段不是点；
        Update(*r+, L, R, c);
    else
    {
        Update(*r, L, a[r].mid(), c);
        Update(*r+, a[r].mid(), R, c);
    }
}
void Query(int r, int L, int R)
{
    if(a[r].c != -)//到达叶子节点；
    {
        if(vis[L] != a[r].c)
            cnt[ a[r].c ]++;
        vis[R] = a[r].c;
        return ;
    }
    if(R-L == )
        return ;
    Query(*r, L, a[r].mid());
    Query(*r+, a[r].mid(), R);
}

int main()
{
    int n, p, q, c;
    while(scanf("%d", &n) != EOF)
    {
        memset(cnt, , sizeof(cnt));
        memset(vis, -, sizeof(vis));

        BuildSegTree(, , );

        while(n--)
        {
            scanf("%d %d %d", &p, &q, &c);
            if(p == q)continue;
            Update(, p, q, c);
        }
        Query(, , );

        for(int i=; i<N; i++)
        {
            if(cnt[i] != )
                printf("%d %d\n", i, cnt[i]);
        }
        printf("\n");
    }
    return ;
}