ZOJ1610_Count the Colors _线段树的树结构

题意

往一个线段上涂色，后涂的会覆盖先前涂的颜色。问涂完后，每一种颜色能看到几段。

思路

用线段树来做。
涂色的过程就是线段树的区间修改。
计数的过程就是单点查询，用 dfs 遍历查看每一个叶子节点的颜色。
计数过程运用了线段树的树形结构的dfs序。

超级坑点

如果两段同色线段之间有一段没涂色的，那么这应该算两段。更新 pre 的时机要注意。WA了n多个小时。

题目链接

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1610

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

const int maxn = 8e3 + 10;

int N = 8000;
int n;
int x, y, z;

int node[maxn << 2];

int ans[maxn];
int pre;

void Push_down(int p)
{
    node[p<<1] = node[p<<1|1] = node[p];
    node[p] = -1;
}

void Add(int p, int l, int r, int x, int y, int z)
{
    if(x <= l && r <= y)
    {
        node[p] = z;
        return;
    }

    if(node[p] != -1) Push_down(p);

    int mid = (l + r) >> 1;
    if(x <= mid) Add(p<<1, l, mid, x, y, z);
    if(y > mid) Add(p<<1|1, mid+1, r, x, y, z);
}

void dfs(int p, int l, int r)
{
    if(l == r)
    {
        if(node[p] != -1 && node[p] != pre)
            ans[node[p]] ++;

        pre = node[p];//就是这里！！！
        return;
    }

    if(node[p] != -1) Push_down(p);

    int mid = (l + r) >> 1;
    dfs(p<<1, l, mid);
    dfs(p<<1|1, mid+1, r);
}

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        memset(node, -1, sizeof node);
        memset(ans, 0, sizeof ans);

        for(int i= 1; i<= n; i++)
        {
            scanf("%d %d %d", &x, &y, &z);

            x ++;
            Add(1, 1, N, x, y, z);
        }

        pre = -1;
        dfs(1, 1, N);

        for(int i= 0; i<= N; i++)
            if(ans[i] > 0) printf("%d %d\n", i, ans[i]);

        printf("\n");
    }

    return 0;
}