C63 可持久化线段树 P3939 数颜色

视频链接：249 可持久化线段树 P3939 数颜色_哔哩哔哩_bilibili

 

 

 

Luogu P3939 数颜色

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 300005
#define mid ((l+r)>>1)
int n,m,a[N];
int root[N],tot; //根节点,开点个数
int ls[N*25],rs[N*25],sum[N*25];
// sum:记录该颜色在各区间出现的次数

void change(int &u,int l,int r,int p,int k){ //点修
  if(!u) u=++tot;
  sum[u]+=k;
  if(l==r) return;
  if(p<=mid) change(ls[u],l,mid,p,k);
  else change(rs[u],mid+1,r,p,k);
}
int query(int u,int l,int r,int x,int y){ //区查
  if(x<=l&&r<=y) return sum[u];
  int s=0;
  if(x<=mid)s+=query(ls[u],l,mid,x,y);
  if(y>mid) s+=query(rs[u],mid+1,r,x,y);
  return s;
}
int main(){
  scanf("%d%d",&n,&m);
  for(int i=1; i<=n; i++){
    scanf("%d",&a[i]);
    change(root[a[i]],1,n,i,1); //建树
  }
  for(int i=1,o,l,r,c,x; i<=m; i++){
    scanf("%d",&o);
    if(o==2){ //交换
      scanf("%d",&x);
      change(root[a[x]],1,n,x,-1);
      change(root[a[x+1]],1,n,x+1,-1);
      change(root[a[x]],1,n,x+1,1);
      change(root[a[x+1]],1,n,x,1);
      swap(a[x],a[x+1]);
    }
    else{ //区查
      scanf("%d%d%d",&l,&r,&c);
      printf("%d\n",query(root[c],1,n,l,r));      
    }
  }
}