线段树+位操作
1. 题意:有一块长为L的长坂,给各个小块涂色,数据给出的要么是A到B上颜色C,要么要求输出A到B有多少种颜色.
2. 其实这道题一看就知道是线段树,而且最多30种颜色,不也难想出用位操作,不过后面一直卡在颜色的更新上,因为如果颜色要更新到叶子节点会超时,如果更新到刚好匹配的区间,对于后来更新的颜色又会有所冲突,要怎么解决呢???
3. 其实方法就是给每个节点加上一个标记cover,如果更新到一个匹配的区间,就把cover赋值为true,这样做有3个用处
(1). 在进行输出dfs时,如果找到的节点的cover是true时,就可以直接返回它的colorstate,因为它就就只有一个颜色
(2). 在后来更新颜色时,如果找到一个点cover为true,就需要先把自己变为false,两个子节点赋值为true,而且把状态赋值给子节点,因为它的子节点的颜色时不正确的,真正的颜色在遇到的第一个cover为true时就被覆盖了,所以要更新子节点,之后在进行正常遍历
(3). 不需要进行buildtree操作.
附代码
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define L 101000
using namespace std;
int state[L<<2];
int n,t,o;
int mi[33];
bool f[L<<2];
int getcolor(int stat)
{
int ret=0;
while(stat)
{
stat&=stat-1;
++ret;
}
return ret;
}
int DFS(int pos,int st,int en,int ll,int rr)
{
if(f[pos])
{
return state[pos];
}
if(st==ll && rr==en)
{
return state[pos];
}
int mid=(st+en)>>1;
if(ll > mid)
{
return DFS(pos<<1|1,mid+1,en,ll,rr);
}
if(rr <= mid)
{
return DFS(pos<<1,st,mid,ll,rr);
}
return DFS(pos<<1,st,mid,ll,mid) | DFS(pos<<1|1,mid+1,en,mid+1,rr);
}
void insert(int pos,int st,int en,int ll,int rr,int color)
{
if(st==ll && en ==rr)
{
state[pos]=mi[color];
f[pos]=true;
return;
}
int mid=(st+en)>>1;
if(f[pos])
{
state[pos<<1]=state[pos];
state[pos<<1|1]=state[pos];
f[pos<<1]=true;
f[pos<<1|1]=true;
}
if(ll>mid)
{
insert(pos<<1|1,mid+1,en,ll,rr,color);
}
else if(rr<=mid)
{
insert(pos<<1,st,mid,ll,rr,color);
}
else
{
insert(pos<<1|1,mid+1,en,mid+1,rr,color);
insert(pos<<1,st,mid,ll,mid,color);
}
state[pos]=state[pos<<1]|state[pos<<1|1];
f[pos]=false;
}
int main()
{
//freopen("input.txt","r",stdin);
mi[1]=1;
for(int i=2;i<32;++i)
{
mi[i]=mi[i-1]<<1;
}
memset(f,true,sizeof(f));
scanf("%d%d%d",&n,&t,&o);
int nn=n<<2;
state[1]=1;
int X,Y,C;
char P;
for(int i=1;i<=o;++i)
{
scanf(" %c ",&P);
if(P =='C')
{
scanf("%d %d %d ",&X,&Y,&C);
if(X>Y) swap(X,Y);
insert(1,1,n,X,Y,C);
}
else
{
scanf("%d %d ",&X,&Y);
if(X>Y) swap(X,Y);
int tt=DFS(1,1,n,X,Y);
printf("%d\n",getcolor(tt));
}
}
return 0;
}