【LOJ6284】数列分块8

题目大意:维护一个序列,支持区间染色,查询区间中等于某个颜色的点的个数。

题解:考虑直接用线段树进行维护,维护区间相同颜色值和一个区间颜色是否相同的标记即可。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10; int n,m,a[maxn];
struct node{
#define ls(x) t[x].lc
#define rs(x) t[x].rc
int lc,rc,val,all;
}t[maxn<<1];
int tot,root;
inline void pushup(int o){
if(t[ls(o)].val==t[rs(o)].val&&t[ls(o)].all&&t[rs(o)].all)t[o].val=t[ls(o)].val,t[o].all=1;
else t[o].all=0;
}
inline void pushdown(int o){
if(t[o].all){
t[ls(o)].all=t[rs(o)].all=1;
t[ls(o)].val=t[rs(o)].val=t[o].val;
}
}
int build(int l,int r){
int o=++tot;
if(l==r){t[o].val=a[l],t[o].all=1;return o;}
int mid=l+r>>1;
ls(o)=build(l,mid),rs(o)=build(mid+1,r);
return pushup(o),o;
}
int query(int o,int l,int r,int x,int y,int c){
if(l==x&&r==y&&t[o].all){
if(t[o].val==c)return r-l+1;
else return t[o].val=c,0;
}
int mid=l+r>>1;
pushdown(o);
int ans=0;
if(y<=mid)ans=query(ls(o),l,mid,x,y,c);
else if(x>mid)ans=query(rs(o),mid+1,r,x,y,c);
else ans=query(ls(o),l,mid,x,mid,c)+query(rs(o),mid+1,r,mid+1,y,c);
return pushup(o),ans;
} int main(){
scanf("%d",&n),m=n;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
root=build(1,n);
while(m--){
int l,r,c;
scanf("%d%d%d",&l,&r,&c);
printf("%d\n",query(root,1,n,l,r,c));
}
return 0;
}
上一篇:获取map集合中key、value


下一篇:把TextView中的文字添加阴影效果及Style的使用 mono版