首页 > 代码库 > POJ 2777 Count Color (线段树+位运算)
POJ 2777 Count Color (线段树+位运算)
题意很简单了,对一个区间有两种操作:
1. "C A B C" Color the board from segment A to segment B with color C.//A~B涂上颜色C
2. "P A B" Output the number of different colors painted between segment A and segment B (including).//输出A~B间颜色的种类数
题目链接:http://poj.org/problem?id=2777
~~~~
注意到颜色的种类只有30种,由此想到用按位或来合并两个区间的颜色(用一个32位的int型,每一位对应一种颜色),
最后求解所得结果有多少个1就是所求的答案了。
剩下的就是线段树的成段更新了。
#include<cstdio> #include<algorithm> #include<cstring> #define lson rt<<1,s,m #define rson rt<<1|1,m+1,e #define N 111111 using namespace std; int tre[N<<2],vis[N<<2]; void pushup(int rt) { tre[rt]=(tre[rt<<1] | tre[rt<<1|1]); } void build(int rt,int s,int e) { tre[rt]=1; vis[rt]=0; if(s==e) return ; int m=(s+e)>>1; build(lson); build(rson); pushup(rt); } void pushdown(int rt) { if(vis[rt]) { vis[rt<<1]=vis[rt<<1|1]=vis[rt]; tre[rt<<1]=tre[rt<<1|1]=tre[rt]; vis[rt]=0; } } int query(int l,int r,int rt,int s,int e) { if(l==s && r==e) return tre[rt]; pushdown(rt); int m=(s+e)>>1; if(r<=m) return query(l,r,lson); else if(l>m) return query(l,r,rson); else return (query(l,m,lson) | query(m+1,r,rson)); } void update(int c,int l,int r,int rt,int s,int e) { if(l<=s && r>=e) { tre[rt]=c; vis[rt]=1; return ; } pushdown(rt); int m=(s+e)>>1; if(r<=m) update(c,l,r,lson); else if(l>m) update(c,l,r,rson); else { update(c,l,m,lson); update(c,m+1,r,rson); } pushup(rt); } int main() { int n,t,m; while(~scanf("%d%d%d",&n,&t,&m)) { build(1,1,n); while(m--) { char op[5]; int l,r,c; scanf("%s",op); if(op[0]=='C') { scanf("%d%d%d",&l,&r,&c); if(l>r) swap(l,r); //~~ update(1<<(c-1),l,r,1,1,n); } else { scanf("%d%d",&l,&r); if(l>r) swap(l,r); //~~ int k=query(l,r,1,1,n); int ans=0; do{ if(k&1) ans++; //求解有多少个1. }while(k>>=1); printf("%d\n",ans); } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。