很显然,这是一道线段树的题,但难点就是怎么保存区间的颜色数,想了很久,曾经考虑过二进制,可是我不太懂二进制,所以就不想了,其实不用二进制也可以,可是脑子笨,不会用数组来保存颜色数,到现在还是不太会,,,,
进入正题吧,这道题颜色最多只有30种,所以可以用二进制的方法来保存每一种不同的颜色,例如 2^0,2^1,2^2,2^4……2^29来表示颜色1,2,3……30.为什么减一呢?因为二进制中2^0表示1,第一位数。怎么表示那颜色数呢?两者位或,然后记录二进制中1的数就是颜色数了,(二进制真是一个强大的工具啊,可是不太懂)。例如2表示10,3表示100,则2|3表示110,有两个1,表示两种颜色。运用位或可以去重(何为位或?例如110|010=110,1010|0110=1110)
用二进制保存颜色数后,剩下的就是线段树的工作了,注意因为是区间更新,所以要用到lazy思想,就是边走边向下传递。。。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define N 100005
int s[N<<2];
int a[N<<2];
void up(int rt)
{
s[rt]=s[rt<<1]|s[rt<<1|1]; //位或
}
void down(int rt)
{
if(a[rt])
{
a[rt<<1]=a[rt<<1|1]=a[rt];
s[rt<<1]=s[rt<<1|1]=1<<(a[rt]-1);
a[rt]=0;
}
}
void build(int l,int r,int rt)
{
a[rt]=0;
if(l==r)
s[rt]=1;
else
{
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
up(rt);
}
}
void updata(int left,int right,int l,int r,int rt,int p)
{
if(left<=l&&right>=r)
{
a[rt]=p;
s[rt]=1<<(p-1);
return ;
}
down(rt);
int m=(l+r)>>1;
if(left<=m)
updata(left,right,l,m,rt<<1,p);
if(right>m)
updata(left,right,m+1,r,rt<<1|1,p);
up(rt);
}
int query(int left,int right,int l,int r,int rt)
{
if(left<=l&&right>=r)
{
return s[rt];
}
down(rt);
int ans=0;
int m=(l+r)>>1;
if(left<=m)
ans|=query(left,right,l,m,rt<<1);
if(right>m)
ans|=query(left,right,m+1,r,rt<<1|1);
return ans;
}
int main()
{
int l,t,o,a,b,c;
char op[10];
scanf("%d%d%d",&l,&t,&o);
build(1,l,1);
while(o--)
{
scanf("%s",op);
if(op[0]=='C')
{
scanf("%d%d%d",&a,&b,&c);
if(a>b)
{
int temp=a;a=b;b=temp;
}
updata(a,b,1,l,1,c);
}
else
{
scanf("%d%d",&a,&b);
if(a>b)
{
int temp=a;a=b;b=temp;
}
//cout<<a<<" "<<b<<endl;
int ans=query(a,b,1,l,1);
//cout<<ans<<endl;
int color=0;
for(int i=0;i<t;i++) //ans中1的个数
if(ans&(1<<i))
color++;
printf("%d\n",color);
}
}
return 0;
}
http://poj.org/problem?id=2777