P2617 Dynamic Rankings-------------------------思维(动态主席树/树套树)

在这里插入图片描述

在这里插入图片描述

解析:
把每个主席树当作成树状数组的一个节点,区间更新的时候,利用树状数组更新,然后在更新到每个版本的主席树

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10000;
int root[N*2],t1[N],t2[N],ls[N*400],rs[N*400],idx;
int a[N],b[N*3],num;
int n,m,n1,n2;
int l[N],r[N],d[N];
int val[N*400];
int lowbit(int x)
{
  return x&(-x);
}
void Add(int &rt,int l,int r,int x,int c)
{
  if(!rt) rt=++idx;
  val[rt]+=c;
  if(l==r) return ;
  int mid=l+r>>1;
  if(x<=mid)  Add(ls[rt],l,mid,x,c);
  else Add(rs[rt],mid+1,r,x,c);

}
void  Change(int pos,int val)
{
  int x=lower_bound(b+1,b+1+num,a[pos])-b;
  for(int i=pos;i<=n;i+=lowbit(i))
    Add(root[i],1,num,x,val);
}
int Kth(int l,int r,int k)
{
  if(l==r) return r;
  int mid=l+r>>1,sum=0;
  for(int i=1;i<=n2;i++) sum+=val[ls[t2[i]]];
  for(int i=1;i<=n1;i++) sum-=val[ls[t1[i]]];

  if(sum>=k)
  {
    for(int i=1;i<=n2;i++) t2[i]=ls[t2[i]];
    for(int i=1;i<=n1;i++) t1[i]=ls[t1[i]];
    return Kth(l,mid,k);
  }
  else
  {
    for(int i=1;i<=n2;i++) t2[i]=rs[t2[i]];
    for(int i=1;i<=n1;i++) t1[i]=rs[t1[i]];
    return Kth(mid+1,r,k-sum);
  }
}
int  Kth_pre(int l,int r,int k)
{
  n1=n2=0;
  for(int i=l-1;i>=1;i-=lowbit(i)) t1[++n1]=root[i];
  for(int i=r;i>=1;i-=lowbit(i)) t2[++n2]=root[i];
  return  Kth(1,num,k);
}
int main()
{
  scanf("%d %d",&n,&m);
  for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[++num]=a[i];
  for(int i=1;i<=m;i++)
  {
    char ch[1000];
    scanf("%s",ch);
    if(*ch=='Q')
    {
      scanf("%d %d %d",&l[i],&r[i],&d[i]);

    }
    else
    {
      scanf("%d %d",&l[i],&d[i]);
      b[++num]=d[i];
    }
  }
  sort(b+1,b+1+num);
  num=unique(b+1,b+1+num)-b-1;
  for(int i=1;i<=n;i++) Change(i,1);
  for(int i=1;i<=m;i++)
  {
    if(r[i])
    {
      printf("%d\n",b[Kth_pre(l[i],r[i],d[i])]);
    }
    else
    {
      Change(l[i],-1);
      a[l[i]]=d[i];
      Change(l[i],1);
    }
  }
}


©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页