POJ2777 - Count Color

第一次看的时候果断跳了，现在写起来发现跳的决定是对的。此题很卡时间，开始建树到叶子果断超时了，后来问了别人改成只对根节点染色才过。后来加上位运算优化才400MS，感觉这种线段树的写法还是很容易挂。

/****************************************************/
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <stack>
#include <map>
#include <queue>
#include <algorithm>
#include <ctime>
#define EPS 1E-8
#define MAXN 100010
#define MAXM 31
#define INF (~0U >> 2)
#define MOD 1000000007
#define Lson l, mid, rt << 1
#define Rson mid + 1, r, rt << 1 | 1
using namespace std;
typedef long long LL;
/****************************************************/
int col[MAXN << 2], vis, res;
void push_down(int rt)
{
	if (col[rt])
	{
		col[rt << 1] = col[rt << 1 | 1] = col[rt];
		col[rt] = 0;
	}
}
void update(int L, int R, int c, int l, int r, int rt)
{
	if (L <= l && r <= R)
	{
		col[rt] = c;
		return;
	}
	push_down(rt);
	int mid = (l + r) >> 1;
	if (L <= mid)
		update(L, R, c, Lson);
	if (R > mid)
		update(L, R, c, Rson);
}
void query(int L, int R, int l, int r, int rt)
{
	if (col[rt])
	{
		if (((1 << col[rt]) & vis) == 0)
			res++;
		vis |= (1 << col[rt]);
		return;
	}
	if (l == r)
		return;
	int mid = (l + r) >> 1;
	if (L <= mid)
		query(L, R, Lson);
	if (R > mid)
		query(L, R, Rson);
}
int main()
{
	int L, T, O;
	while (scanf("%d%d%d", &L, &T, &O) != EOF)
	{
		memset(col, 0, sizeof(col));
		col[1] = 1;
		char str[10];
		while (O--)
		{
			scanf("%s", str);
			if (str[0] == 'C')
			{
				int a, b, c;
				scanf("%d%d%d", &a, &b, &c);
				if (a > b)
					swap(a, b);
				update(a, b, c, 1, L, 1);
			}
			else if (str[0] == 'P')
			{
				int a, b;
				scanf("%d%d", &a, &b);
				if (a > b)
					swap(a, b);
				res = vis = 0;
				query(a, b, 1, L, 1);
				printf("%d\n", res);
			}
		}
	}
}