POJ 2777 Count Color

题意：和贴海报这个题类似……只是多了一个查询l到r区间的操作

解法：相对贴海报来说不需要离散化，查询操作需要稍作改动。

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<math.h>
#include<limits.h>
#include<time.h>
#include<stdlib.h>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#define LL long long
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

const int MAXN = 100005;
int col[MAXN << 2];
bool flag[35];
int ans;
using namespace std;

void pushdown(int rt)
{
    if(col[rt] != -1)
    {
        col[rt << 1] = col[rt << 1 | 1] = col[rt];
        col[rt] = -1;
    }
}
void update(int ll, int rr, int c, int l, int r, int rt)
{
    if(ll <= l && rr >= r)
    {
        col[rt] = c;
        return ;
    }
    pushdown(rt);
    int m = (l + r) >> 1;
    if(ll <= m) update(ll, rr, c, lson);
    if(rr > m) update(ll, rr, c, rson);
}
void query(int ll, int rr, int l, int r, int rt)
{
    if(col[rt] != -1)
    {
        flag[col[rt]] = true;
        return ;
    }
    int m = (l + r) >> 1;
    if(ll <= m) query(ll, rr, lson);
    if(rr > m) query(ll, rr, rson);
}
int main()
{
    int n, t, o;
    while(~scanf("%d%d%d", &n, &t, &o))
    {
        memset(col, -1, sizeof(col));
        update(1, n, 1, 1, n, 1);
        while(o--)
        {
            char ch[2];
            int l, r;
            scanf("%s%d%d", ch, &l, &r);
            if(ch[0] == 'C')
            {
                int c;
                scanf("%d", &c);
                update(l, r, c, 1, n, 1);
            }
            else
            {
                memset(flag, 0, sizeof flag);
                query(l, r, 1, n, 1);
                int ans = 0;
                for(int i = 1; i <= t; i++)
                    ans += flag[i];
                printf("%d
", ans);
            }
        }
    }
    return 0;
}