首页 > 代码库 > PKU 2777 Count Color (线段树区间更新)
PKU 2777 Count Color (线段树区间更新)
题意: 给你三个数:L (1 <= L <= 100000), T (1 <= T <= 30) and O (1 <= O <= 100000),表示有一长度为L的板(1~L),
有T种颜色(1~T),然后有O个操作,初始板1~L的颜色为1,"C A B C"表示在区间A,B图上C颜色, "P A B" 表示询问
A,B区间有几种不同的颜色。
#include <stdio.h> #include <iostream> #include <algorithm> #include <string.h> #include <math.h> #define M 100000 #define LL long long using namespace std; struct Tree { int l,r,c;//c代表颜色,若区间(l,r)的颜色不止一种则c=-1; }tree[M*3]; int color[35];//在查询时,若[A,B]区间内出现过某种颜色k,则color[k]=true; void build(int id,int l,int r) { tree[id].l=l; tree[id].r=r; tree[id].c=1;//初始颜色为1 if(l==r) return ; build(id*2,l,(l+r)/2); build(id*2+1,(l+r)/2+1,r); } void update(int id,int l,int r,int co) { if(tree[id].l>=l&&tree[id].r<=r)//(l,r)完全覆盖id节点 { tree[id].c=co; return; } if(tree[id].c>0)//一直wa在这里 { tree[id*2].c=tree[id*2+1].c=tree[id].c; tree[id].c=-1; } tree[id].c=-1;//没有完全覆盖 int mid=(tree[id].l+tree[id].r)/2; if(mid>=r) update(2*id,l,r,co); else if(mid<l) update(2*id+1,l,r,co); else { update(2*id,l,mid,co); update(2*id+1,mid+1,r,co); } } void P(int id,int l,int r) { if(tree[id].c>0) { color[tree[id].c]=true; return; } if(tree[id].l==tree[id].r) return; int mid=(tree[id].l+tree[id].r)/2; if(mid>=r) P(id*2,l,r); else if(mid<l) P(id*2+1,l,r); else { P(id*2,l,mid); P(id*2+1,mid+1,r); } } int Ans(int t) { int ans=0; for(int i=1;i<=t;i++) if(color[i]) ans++; return ans; } int n,t,m; int main() { scanf("%d%d%d",&n,&t,&m); char s[2]; int a,c,b; build(1,1,n); while(m--) { scanf("%s",s); if(s[0]=='C') { scanf("%d%d%d",&a,&b,&c); if(a>b) swap(a,b); update(1,a,b,c); } else { scanf("%d%d",&a,&b); if(a>b) swap(a,b); memset(color,false,sizeof(color)); P(1,a,b); printf("%d\n",Ans(t)); } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。