POJ 2777 Count Color(线段树:区间覆盖)

POJ 2777 Count Color(线段树:区间覆盖)

http://poj.org/problem?id=2777

题意：

        题意：有一个长板子，多次操作，有两种操作，第一种是给从a到b那段染一种颜色c，另一种是询问a到b有多少种不同的颜色。

分析:

本题是POJ 1436的弱化版本:

http://blog.csdn.net/u013480600/article/details/22313189

       首先本题的区间是一个一个的整数,所以不用乘2考虑开区间闭区间的问题了.所以这里的线段树维护一个color信息,其中color>0表示节点所指的区间都有某种颜色color, color=-1 表示该节点的子节点的颜色不一致.

       build操作: 递归建树,并且所有color初始设置为1.

       PushUp操作: 根据子节点的color更新i节点的color.如果子节点color一致,那么i节点color与子节点color一样.否则i节点color为-1.

       PushDown操作: color>0时才把color传递到下面去.PushDown操作可以不用改变i(父)节点的color.

       update操作: 如果[ql, qr]包括了[l, r] ,那么就直接置新color值,否则先PushDonw,然后分段更新color,再PushUp.

       query操作: 本操作主要是用一个cnt全局变量反馈[ql, qr]区间内一共有多少个不同的color.每次查询前都初始化vis[30]为false,出现一种颜色就vis[i]=true.如果[ql, qr]包括了[l, r]区间,那么直接更新vis,否则PushDown后,分段查找.

AC代码:329ms

<span style="font-size:18px;">//一次AC
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
const int MAXN = 100000 + 100;
bool vis[35];
int cnt;//用来计数颜色
struct IntervalTree
{
    int color[MAXN * 4];

    void build(int i, int l, int r)
    {
        color[i] = 1;
        if(l == r) return ;
        int m = (l + r) / 2;
        build(lson);
        build(rson);
    }

    void PushDown(int i)
    {
        if(color[i] > 0)
            color[i * 2] = color[i * 2 + 1] = color[i];
    }

    void PushUp(int i)
    {
        if(color[i * 2] == -1 || color[i * 2 + 1] == -1)
            color[i] = -1;
        else if(color[i * 2] == color[i * 2 + 1])
            color[i] = color[i * 2];
        else
            color[i] = -1;
    }

    void update(int ql, int qr, int v, int i, int l, int r)
    {
        if(ql <= l && r <= qr)
        {
            color[i] = v;
            return ;
        }
        PushDown(i);
        int m = (l + r) / 2;
        if(ql <= m) update(ql, qr, v, lson);
        if(m < qr) update(ql, qr, v, rson);
        PushUp(i);
    }

    void query(int ql, int qr, int i, int l, int r)
    {
        if(color[i] > 0)
        {
            if(vis[color[i]] == false)
                cnt++;
            vis[color[i]] = true;
            return ;
        }
        //PushDown(i);
        int m = (l + r) / 2;
        if(ql <= m) query(ql, qr, lson);
        if(m < qr) query(ql, qr, rson);
    }
};
IntervalTree T;
int main()
{
    int n, t, q;
    while(scanf("%d%d%d", &n, &t, &q) == 3)
    {
        T.build(1, 1, n);
        while(q--)
        {
            char str[10];
            scanf("%s", str);
            if(str[0] == 'C')
            {
                int x, y, z;
                scanf("%d%d%d", &x, &y, &z);
                T.update(x, y, z, 1, 1, n);
            }
            else if(str[0] == 'P')
            {
                int x, y;
                scanf("%d%d", &x, &y);
                memset(vis, 0, sizeof(vis));
                cnt = 0;
                T.query(x, y, 1, 1, n);
                printf("%d\n", cnt);
            }
        }
    }
    return 0;
}
</span>