Count Color(poj 2777)
题意:
给一个固定长度为L的画板
有两个操作:
C A B C:区间AB内涂上颜色C。
P A B:查询区间AB内颜色种类数。
分析:显然是要用线段树来操作的,设定一个sum[]来维护一个区间内的颜色,若为-1,则说明不是叶子节点,还应在往底下找;
同时在查询时维护一个vis[]来记录在这个区间内哪个颜色被使用过了。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include<cstdio> #include<iostream> #include<cstring> #define lson l,m,now*2 #define rson m+1,r,now*2+1 #define M 400010 #define N 35 using namespace std; int vis[N],sum[M];//当sum[i]=-1是,说明不是底下至少有2种颜色,否则sum[i]是i的颜色 void build(int l,int r,int now) { sum[now]=1; if(l==r)return; int m=(l+r)/2; build(lson); build(rson); } void push_down(int now) { sum[now*2]=sum[now]; sum[now*2+1]=sum[now]; sum[now]=-1; } void change(int x,int y,int v,int l,int r,int now) { if(l>=x&&r<=y) { sum[now]=v; return; } if(sum[now]==v)return; if(sum[now]!=-1)push_down(now); int m=(l+r)/2; if(x<=m)change(x,y,v,lson); if(y>m)change(x,y,v,rson); } void query(int x,int y,int l,int r,int now) { if(sum[now]!=-1) { vis[sum[now]]=1; return; } int m=(l+r)/2; if(x<=m)query(x,y,lson); if(y>m)query(x,y,rson); } int main() { int n,m,p; scanf("%d%d%d",&n,&m,&p); build(1,n,1); for(int i=1;i<=p;i++) { char c; cin>>c; if(c=='C') { int a,b,v; scanf("%d%d%d",&a,&b,&v); if(a>b)swap(a,b); change(a,b,v,1,n,1); } else { memset(vis,0,sizeof(vis)); int a,b,tot=0; scanf("%d%d",&a,&b); if(a>b)swap(a,b); query(a,b,1,n,1); for(int i=1;i<=30;i++) if(vis[i])tot++; printf("%d\n",tot); } } return 0; }