POJ 2777 Count Color(线段树+位运算)

来源：互联网发布：大数据泄露编辑：程序博客网时间：2024/06/06 12:28

更新的时候要用延迟标记,用位运算保存儿子结点合并的结果,强大啊.

#include <iostream>#include <memory.h>#include <cstdio>using namespace std;#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1const int maxn=100010;int segs[maxn*3],L,T,O;bool f[maxn*3];char buf[5];void build(int l,int r,int rt){segs[rt]=1;f[rt]=0;if(l==r)return;int m=(l+r)>>1;build(lson),build(rson);}void pushDown(int rt){if(f[rt]){f[rt<<1]=f[rt<<1|1]=f[rt];segs[rt<<1]=segs[rt];segs[rt<<1|1]=segs[rt];f[rt]=0;}}void update(int C,int L,int R,int l,int r,int rt){if(L<=l&&r<=R){segs[rt]=C;f[rt]=1;return;}pushDown(rt);int m=(l+r)>>1;if(L<=m)update(C,L,R,lson);if(R>m)update(C,L,R,rson);segs[rt]=segs[rt<<1]|segs[rt<<1|1];}int query(int L,int R,int l,int r,int rt){if(L<=l&&r<=R){return segs[rt];}pushDown(rt);int m=(l+r)>>1,ret=0;if(L<=m)ret|=query(L,R,lson);if(R>m)ret|=query(L,R,rson);return ret;}int main(){scanf("%d%d%d",&L,&T,&O);build(1,L,1);while (O--){scanf("%s",buf);if(buf[0]=='C'){int A,B,C;scanf("%d%d%d",&A,&B,&C);if(A>B)swap(A,B);update(1<<(C-1),A,B,1,L,1);}else {int A,B;scanf("%d%d",&A,&B);if(A>B)swap(A,B);int ans=0,ret=query(A,B,1,L,1);for (int i=0;ret;ret>>=1){if(1&ret)ans++;}printf("%d\n",ans);}}return 0;}