2019年ACM-ICPC - 南昌网络赛I：Yukino With Subinterval【带修主席树】

题目:

2019ICPC南昌网络赛I：Yukino With Subinterval

题意：

给定长度为 N 的数组，有两种操作：（1）单点修改，（2）查询区间【qL，qR】内有多少个不同的段(连续相同的为一段)，且满足每一段的数字介于【x，y】之间

分析：

为了补这个题，专门学了带修主席树~~~

先把每一段的数字压缩到它的左端点上，其它位置为0，那么（2）操作相当于查询区间 [qL,qR] 内属于 [x,y] 的数字有多少个（当然要特判一下 qL 的情况），这是主席树的简单应用；对于（1）操作，修改后会影响 x 和 x+1，主席树单点修改即可；注意修改操作没写好极容易消耗更多的内存，导致MLE或段错误

#include <bits/stdc++.h>
 
#define lowbit(x) (x)&(-x)
using namespace std;
const int maxn = 2e5+25;
int n,m,tot,a[maxn],c[maxn],root[maxn],bit[maxn],op,l,r,x,y;
struct tree{
    int l,r,sum;
}T[maxn*150];
void updata(int l,int r,int &x,int y,int val,int w,int f){
    if(f){T[++tot] = T[y]; T[tot].sum += w; x = tot;}
    else{if(!x) x = ++tot; T[x].sum += w;}           //树状数组不需要可持久化，这样写能省很多空间
    if(l == r) return;
    int mid = (l+r) >> 1;
    if(val > mid) updata(mid+1,r,T[x].r,T[y].r,val,w,f);
    else updata(l,mid,T[x].l,T[y].l,val,w,f); 
}
void change(int x,int w){
    int val = c[x];
    while(x <= n){
        updata(1,n,bit[x],bit[x],val,w,0);
        x += lowbit(x);
    }
}
int persum(int x,int t,int nxt[][2]){
    int res = 0,k = 0;
    while(x > 0){
        res += T[nxt[k++][t]].sum;
        x -= lowbit(x);
    }
    return res;
}
void query(int l,int r,int L,int R,int ql,int qr,int x,int y,int &ans,int nxt[][2]){
    if(l > R || r < L) return;
    if(l >= L && r <= R){
        ans += T[y].sum-T[x].sum+persum(qr,0,nxt)-persum(ql,1,nxt);
        return ;
    }
    int mid = (l+r)>>1;int nxt1[25][2] = {0},k0 = 0,k1 = 0;
    for(int i = qr;i > 0;i -= lowbit(i),k0++) nxt1[k0][0] = T[nxt[k0][0]].l;
    for(int i = ql;i > 0;i -= lowbit(i),k1++) nxt1[k1][1] = T[nxt[k1][1]].l;
    query(l,mid,L,R,ql,qr,T[x].l,T[y].l,ans,nxt1);
     k0 = 0,k1 = 0; 
    for(int i = qr;i > 0;i -= lowbit(i),k0++) nxt1[k0][0] = T[nxt[k0][0]].r;
    for(int i = ql;i > 0;i -= lowbit(i),k1++) nxt1[k1][1] = T[nxt[k1][1]].r;
    query(mid+1,r,L,R,ql,qr,T[x].r,T[y].r,ans,nxt1);
}
int nxt[25][2];
int main(){
    scanf("%d %d",&n,&m);
    for(int i = 1;i <= n; ++i) scanf("%d",a+i);
    for(int i = 1;i <= n; ++i){
        if(a[i] != a[i-1]) updata(1,n,root[i],root[i-1],a[i],1,1),c[i] = a[i];
        else root[i] = root[i-1],c[i] = 0;
    }
    while(m--){
        scanf("%d",&op);
        if(op == 1){
            scanf("%d %d",&x,&y);
            if(a[x] == y) continue;
            a[x] = y;
            if(c[x] != 0) change(x,-1);
            if(a[x] != a[x-1]){
                c[x] = a[x]; change(x,1);
            }
            else c[x] = 0;
            if(x+1 > n) continue;
            if(c[x+1]!=0&&a[x+1]==a[x]) change(x+1,-1),c[x+1] = 0;
            if(c[x+1]==0&&a[x+1]!=a[x]) c[x+1]=a[x+1],change(x+1,1);
        }else{
            scanf("%d %d %d %d",&l,&r,&x,&y);
            int ans = 0,k0 = 0,k1 = 0;
            for(int i = r;i > 0;i -= lowbit(i)) nxt[k0++][0] = bit[i];
            for(int i = l-1;i > 0;i-=lowbit(i)) nxt[k1++][1] = bit[i];
            query(1,n,x,y,l-1,r,root[l-1],root[r],ans,nxt);
            if(c[l]==0&&a[l]>=x&&a[l]<=y) ans++;    //特判左端点
            printf("%d\n",ans);
        }
    }
    return 0;
}