POJ2777 区间更新+区间询问（线段树）

   此题与前两题不同。题意是将不同区间覆盖成不同颜色，在查询区间的时候询问共有多少种不同的颜色。

   大体思路是这样的：

        在结构体内记录当前树的双子节点，所染的颜色color，以及flag标记。在更新时，用flag来标记当前节点是否为最新一个被更新的节点，并且更新color值。在查询或更新时，flag可以看成为懒惰标记，如果存在的话，变将color和flag标记往下传，并将当前节点的flag赋值为false。

        在查询的时候，如果当前节点有标记，那么用trag[tree[x].color]判断该颜色是否被计算过，如果没有，ans++。

        当然了，一个注意点就是每次查询前，需要对trag数组memset一下。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
const int maxn = 1e6+5;
using namespace std;
int n,l,o,t,a,b,c,ans;
bool trag[35];
struct node{
	int l,r,color;
	bool flag;
}tree[maxn*4];

void push_down(int x)
{
	if(tree[x].flag)
	{
		tree[x<<1].color = tree[x].color;
		tree[x<<1|1].color = tree[x].color;
		tree[x<<1].flag = tree[x<<1|1].flag = 1;
		tree[x].flag = 0;
	}
}

void build(int x,int l,int r)
{
	tree[x].l = l;
	tree[x].r = r;
	tree[x].color = tree[x].flag = 1;
	if(l==r) return;
	else{
		int mid = (l+r)/2;
		build(x<<1,l,mid);
		build(x<<1|1,mid+1,r);
	}	
}

void update(int x,int l,int r,int val)
{
	int L = tree[x].l, R = tree[x].r;
	if(l<=L&&r>=R) {
		tree[x].color = val;
		tree[x].flag = 1;
		return; 
	}
	push_down(x);
	int mid = (L+R)/2;
	if(l<=mid)
	{
		update(x<<1,l,r,val);
	}
	if(r>mid)
	 update(x<<1|1,l,r,val);
}

int query(int x,int l,int r)
{
	int L = tree[x].l, R = tree[x].r;
	//cout<<tree[x].flag<<" xxxxx"<<endl;
	//cout<<L<<" "<<R<<"xxxxxxxxx"<<endl;
	long long ans = 0;
	if(tree[x].flag) {
		if(!trag[tree[x].color])
		{
			ans++;
			trag[tree[x].color] = 1;
		}
		return ans;
	}	
	int mid = (L+R)/2;
	if(l<=mid)
	 ans += query(x<<1,l,r);
	if(r>mid)
	 ans += query(x<<1|1,l,r);
	return ans;
}

int main()
{
	scanf("%d%d%d",&n,&t,&o);
	build(1,1,n);
	getchar();
	for(int i=0;i<o;i++)
	{
		string s;
		cin>>s;
		if(s[0]=='C'){
			scanf("%d%d%d",&a,&b,&c);
			if(a>b) swap(a,b);
			update(1,a,b,c);
		}		
		else{
			scanf("%d%d",&a,&b);
			memset(trag,0,sizeof(trag));
			if(a>b) swap(a,b);
			ans = query(1,a,b);
			cout<<ans<<endl;
		}
		getchar();
	}
	return 0;
}