pku 2777 Count Color(线段树变形)

 

因为操作次数很多，所以修改操作应该选用O(logN)的方式，线段树成为最佳选择。
这题有两个注意的地方，一是因为线段以segment为划分单位，所以右子树的left比左子树的right要大1。二是给线段树结点增加一个color域，当整个线段为纯色时，color=颜色代号，当线段内有多种颜色是，color=-1。这里当父节点涂色时，说明其所有子节点的颜色都需要更新。但因为考虑到效率问题，不可能去这样操作，这样就给insert操作和cal操作带来了一点改变，需要特别注意(通常线段树都是子节点控制父节点，这次是父节点控制子节点)。
 

#include <iostream>
using namespace std;
#define MAXLENGTH 100005
struct node
{
int left,right,color;
}my_tree[MAXLENGTH*10];
bool used[30+5];
void swap(int a,int b)
{
int temp=a;
a=b;
b=temp;
}
void build(int a,int b,int i)
{
my_tree[i].left=a;
my_tree[i].right=b;
if(b-a)//当segment大于2时，就继续往下划分
{
build(a,(a+b)/2,i*2);
build((a+b)/2+1,b,i*2+1);
}
}
void insert(int a,int b,int c,int i)
{
if(my_tree[i].left==a&&my_tree[i].right==b)
{
my_tree[i].color=c;
}
else
{
if(my_tree[i].color>0) my_tree[i*2].color=my_tree[i*2+1].color=my_tree[i].color,my_tree[i].color=-1;//极为关键
int mid=(my_tree[i].left+my_tree[i].right)/2;
if(b<=mid)
{
insert(a,b,c,2*i);
}
else if(a>mid)
{
insert(a,b,c,2*i+1);
}
else
{
insert(a,mid,c,i*2);
insert(mid+1,b,c,i*2+1);
}

}
}
void cal(int a,int b,int i)
{
if(my_tree[i].color>0) //正确的操作下，总是有left<=a<=b<=right，所以只要线段颜色确定，立即返回
{
used[my_tree[i].color]=true;
}
else
{
int mid=(my_tree[i].left+my_tree[i].right)/2;
if(b<=mid) cal(a,b,2*i);
else if(a>mid) cal(a,b,2*i+1);
else
{
cal(a,mid,i*2);
cal(mid+1,b,i*2+1);
}
}
}
int main()
{
int L,T,O;
char flag;
int A,B,C,ans;
scanf("%d%d%d",&L,&T,&O);
build(1,L,1);
my_tree[1].color=1;
for(int i=0;i<O;i++)
{
scanf(" %c",&flag);
if(flag=='C')
{
scanf("%d%d%d",&A,&B,&C);
if(A>B) swap(A,B);
insert(A,B,C,1);
}
else
{
scanf("%d%d",&A,&B);
if(A>B) swap(A,B);
memset(used,0,sizeof(used));
cal(A,B,1);
ans=0;
for(int j=1;j<=T;j++)
{
if(used[j]) ans++;
}
printf("%d/n",ans);
}
}
return 0;
}