【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;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10612875.html