Poj 2777 Count Color

题目大意：给你一个长度为n的线段（n<1000000)，还有t种颜料（t<30）。有以下两种操作：

C a b c :将从a到b的线段涂上c色；

P a b:询问从a到b的线段上有多少种颜色。

思路：线段树的应用，另外还用到了一个线段树常用的思路——lazy。lazy顾名思义就是懒惰——这里是指在更新的时候，不必更新到每条元线段。

#include <iostream>
using namespace std;
#include <stdio.h>
#include <string.h>
#include <memory.h>
const int MAXN = 100005;
const int COLOR_NUM=35;
const int DEFAULT_COLOR=0;

struct node {
int left;
int right;
int color;
};
node tree[MAXN*5];
bool visit[COLOR_NUM];
int l;

void build(const int &left, const int &right, const int &node_num=1) {
tree[node_num].left=left;
tree[node_num].right=right;
tree[node_num].color=DEFAULT_COLOR;

if (left==right)
return;
int mid = (left+right) >> 1;
build(left,mid,2*node_num);
build(mid+1,right,2*node_num+1);
}
void init() {
build(1,l);
tree[1].color=1;
}
void insert(const int &left, const int &right, const int &color, const int &node_num = 1 ) {
if (left==tree[node_num].left&&right==tree[node_num].right) {
tree[node_num].color=color;
return;
}
if (tree[node_num].color!=DEFAULT_COLOR) {
tree[node_num*2].color=tree[node_num].color;
tree[node_num*2+1].color=tree[node_num].color;
tree[node_num].color=DEFAULT_COLOR;
}
int mid=(tree[node_num].left+tree[node_num].right) >> 1;
if (right<=mid)
insert(left,right,color,node_num*2);
else if (left>mid)
insert(left,right,color,node_num*2+1);
else {
insert(left,mid,color,node_num*2);
insert(mid+1,right,color,node_num*2+1);
}
}

int query(const int &left, const int &right, const int &node_num=1) {
int ctr=0;
if (tree[node_num].color!=DEFAULT_COLOR) {
if (visit[tree[node_num].color]==false) {
ctr++;
visit[tree[node_num].color]=true;
}
return ctr;
}
int mid = (tree[node_num].left+tree[node_num].right) >> 1;
if (right<=mid)
ctr+=query(left,right,node_num*2);
else if (left>mid)
ctr+=query(left,right,node_num*2+1);
else {
ctr+=query(left,mid,node_num*2);
ctr+=query(mid+1,right,node_num*2+1);
}
return ctr;
}
int main()
{
int t,o,i;
char operation;
int a,b,c,temp;

scanf("%d%d%d",&l,&t,&o);
init();

while (o--) {
getchar();
scanf("%c",&operation);
if (operation=='C') {
scanf("%d%d%d",&a,&b,&c);
if (a>b) {
temp=a;
a=b;
b=temp;
}
insert(a,b,c);
}
else {
scanf("%d%d",&a,&b);
if (a>b) {
temp=a;
a=b;
b=temp;
}
memset(visit,false,sizeof(visit));
printf("%d\n",query(a,b));
}
}
return 0;
}