1. 程式人生 > >poj 2777 Count Color

poj 2777 Count Color

圖片 pushd const poj string name scan tdi mes

和hdu5023一樣不過詢問不同。

技術分享圖片
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <iostream>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define up rt,rt<<1,rt<<1|1
using namespace std;
const int M = 1e6+7;
int n,q;
int c[M<<2],ans[20000],cnt; void pushdown(int rt,int l,int r){ c[l]=c[r]=c[rt];c[rt]=0; } void build(int l,int r,int rt){ c[rt]=0; if(l==r) return ; int mid=(l+r)>>1; build(lson); build(rson); } void update(int L,int R,int l,int r,int rt,int v){ if(L==l&&r==R){ c[rt]
=v;//這個區間是整個被染色 return ; } if(c[rt]) pushdown(up);//當這個區間不是整個會被染色那麽將顏色下放 int mid=(l+r)>>1; if(L<=mid) update(L,min(mid,R),lson,v); if(R>mid) update(max(L,mid+1),R,rson,v); } void query(int L,int R,int l,int r,int rt){ //cout<<rt<<endl; if(l<=L&&R<=r&&c[rt]){//
查詢區間整個被這個區間覆蓋且這個區間有顏色就返回顏色 ans[++cnt]=c[rt]; return ; } int mid=(l+r)>>1; if(L<=mid) query(L,min(mid,R),lson); if(R>mid) query(max(mid+1,L),R,rson); } int main(){ int _; while(~scanf("%d%d%d",&n,&_,&q)){ build(1,n,1); c[1]=1;cnt=0;memset(ans,0,sizeof(ans)); while(q--){ char s[10];int x,y,v; scanf("%s",s); if(s[0]==C){ scanf("%d%d%d",&x,&y,&v);if(x>y) swap(x,y); update(x,y,1,n,1,v); } else{ cnt=0;memset(ans,0,sizeof(ans)); scanf("%d%d",&x,&y);if(x>y) swap(x,y); query(x,y,1,n,1); sort(ans+1,ans+cnt+1); int sz=unique(ans+1,ans+cnt+1)-ans-1;//去重,因為可能找到很多重復染色區間 printf("%d\n",sz); } } } return 0; }
View Code

poj 2777 Count Color