C105 整体二分+树状数组 P2617 Dynamic Rankings

视频链接:C105 整体二分+树状数组 P2617 Dynamic Rankings_哔哩哔哩_bilibili

 

 

C96 树状数组套权值线段树 P2617 Dynamic Rankings - 董晓 - 博客园 (cnblogs.com)

C104【模板】整体二分+树状数组 P3834 可持久化线段树2 - 董晓 - 博客园 (cnblogs.com)

Luogu P2617 Dynamic Rankings

复制代码
// 整体二分+树状数组 O(n*logn*logV)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=300005;
int n,m,cnt,a[N];
int ans[N],s[N];

struct Q{
  //  数: x位置,y值,k贡献,opt=0  
  //查询: [x,y]第k小,id编号,opt=1
  int x,y,k,id,opt;
}q[N<<1],q1[N<<1],q2[N<<1];

void add(int x,int v){ //加入贡献
  while(x<=n)s[x]+=v,x+=x&(-x);
}
int sum(int x){ //前缀和
  int t=0;
  while(x)t+=s[x],x-=x&(-x);
  return t;
}
void solve(int l,int r,int L,int R){
  if(l>r) return; //[l,r]数据个数域 [L,R]值域
  if(L==R){
    for(int i=l;i<=r;i++)
      if(q[i].opt) ans[q[i].id]=L; //记录答案
    return;
  }
  int mid=(L+R)>>1,p1=0,p2=0;
  for(int i=l;i<=r;i++){
    if(!q[i].opt){          //若是数,按值分流
      if(q[i].y<=mid)   
        add(q[i].x,q[i].k), //加入贡献
        q1[++p1]=q[i];      //分流到左边
      else q2[++p2]=q[i];   //分流到右边
    }
    else{                   //若是查询,按个数分流
      int s=sum(q[i].y)-sum(q[i].x-1);
      if(s>=q[i].k) q1[++p1]=q[i];  //分流到左边
      else q[i].k-=s,q2[++p2]=q[i]; //分流到右边
    }
  }
  for(int i=1;i<=p1;i++)
    if(!q1[i].opt) add(q1[i].x,-q1[i].k);  //减去贡献
  for(int i=1;i<=p1;i++)q[i+l-1]=q1[i]; 
  for(int i=1;i<=p2;i++)q[i+l+p1-1]=q2[i]; //合并数组
  solve(l,l+p1-1,L,mid);
  solve(l+p1,r,mid+1,R); //分治
}
int main(){
  scanf("%d%d",&n,&m); char ch[2]; int x,y,k;
  for(int i=1;i<=n;i++)
    scanf("%d",&a[i]),q[++cnt]={i,a[i],1,0,0};
  for(int i=1;i<=m;i++){
    scanf("%s%d%d",ch,&x,&y);
    if(ch[0]=='C')
      q[++cnt]={x,a[x],-1,0,0},  //旧数贡献-1
      q[++cnt]={x,a[x]=y,1,0,0}; //新数贡献+1
    else scanf("%d",&k),q[++cnt]={x,y,k,i,1};
  }
  memset(ans,-1,sizeof ans);
  solve(1,cnt,0,1e9); //整体二分
  for(int i=1;i<=m;i++)if(~ans[i])printf("%d\n",ans[i]);
}
复制代码

 

复制代码
// 整体二分+树状数组 O(nlognlogV)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=300005;
int n,m,cnt,a[N];
int ans[N],s[N];

struct Q{
  //查询: [x,y]第k小,id编号,opt=1
  //  数: x位置,y值,k贡献,opt=0
  int x,y,k,id,opt;
}q[N<<1],t[N<<1];

void add(int x,int v){ //加入贡献
  while(x<=n)s[x]+=v,x+=x&(-x);
}
int sum(int x){ //前缀和
  int t=0;
  while(x)t+=s[x],x-=x&(-x);
  return t;
}
void solve(int l,int r,int L,int R){
  if(l>r) return; //[l,r]数据个数域 [L,R]值域
  if(L==R){
    for(int i=l;i<=r;i++)
      if(q[i].opt) ans[q[i].id]=L; //记录答案
    return;
  }
  int mid=(L+R)>>1,p1=l-1,p2=r+1;
  for(int i=l;i<=r;i++){
    if(!q[i].opt){          //若是数,按值分流
      if(q[i].y<=mid)   
        add(q[i].x,q[i].k), //加入贡献
        t[++p1]=q[i];      //分流到左边
      else t[--p2]=q[i];   //分流到右边
    }
    else{                   //若是查询,按个数分流
      int s=sum(q[i].y)-sum(q[i].x-1);
      if(s>=q[i].k) t[++p1]=q[i];  //分流到左边
      else q[i].k-=s,t[--p2]=q[i]; //分流到右边
    }
  }
  for(int i=l;i<=p1;i++)
    if(!t[i].opt) add(t[i].x,-t[i].k);  //减去贡献
  for(int i=l;i<=p1;i++)q[i]=t[i]; 
  for(int i=p2;i<=r;i++)q[i]=t[r-i+p2]; //合并数组
  solve(l,p1,L,mid);
  solve(p2,r,mid+1,R); //分治
}
int main(){
  scanf("%d%d",&n,&m); char ch[2]; int x,y,k;
  for(int i=1;i<=n;i++)
    scanf("%d",&a[i]),q[++cnt]={i,a[i],1,0,0};
  for(int i=1;i<=m;i++){
    scanf("%s%d%d",ch,&x,&y);
    if(ch[0]=='C')
      q[++cnt]={x,a[x],-1,0,0},  //旧数贡献-1
      q[++cnt]={x,a[x]=y,1,0,0}; //新数贡献+1
    else scanf("%d",&k),q[++cnt]={x,y,k,i,1};
  }
  memset(ans,-1,sizeof ans);
  solve(1,cnt,0,1e9); //整体二分
  for(int i=1;i<=m;i++)if(~ans[i])printf("%d\n",ans[i]);
}
复制代码

 

复制代码
// 整体二分+树状数组 O(nlognlogV)
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N=300005;
#define lowbit(x) (x&-x)
int n,m,a[N],ans[N];

struct Q{
  //查询: [x,y]第k小,id编号,opt=1
  //原数: x位置,y值,opt=0
  int x,y,k,id,opt;
}; 
vector<Q>q; //数据序列

struct BIT{
  vector<int>s;
  void init(int n){s.resize(n);}
  void add(int x,int k){
    while(x<=n)s[x]+=k,x+=lowbit(x);
  }
  int sum(int x){
    int t=0;
    while(x)t+=s[x],x-=lowbit(x);
    return t;
  }
  int sum(int l,int r){
    return sum(r)-sum(l-1);
  }
}tree; //树状数组

void solve(vector<Q>q,int L,int R){
  if(!q.size()) return;
  if(L==R){
    for(auto i:q) if(i.opt) ans[i.id]=L;
    return; 
  }
  int mid=L+R>>1; vector<Q>q1,q2;
  for(auto i:q){
    if(!i.opt){ //若是原数,按值分流
      if(i.y<=mid)
        tree.add(i.x,i.k),    //加入贡献
        q1.push_back(i);    //分流到左边
      else q2.push_back(i); //分流到右边
    }
    else{      //若是查询,按个数分流
      int s=tree.sum(i.x,i.y);    
      if(s>=i.k) q1.push_back(i); //分流到左边
      else i.k-=s,q2.push_back(i);//分流到右边
    }
  }
  for(auto i:q1)
    if(!i.opt) tree.add(i.x,-i.k); //减去贡献
  solve(q1,L,mid); 
  solve(q2,mid+1,R); //分治
}
int main(){
  scanf("%d%d",&n,&m); char ch[2]; int x,y,k;
  tree.init(n+1);
  for(int i=1;i<=n;i++)
    scanf("%d",&a[i]),q.push_back({i,a[i],1,0,0});
  for(int i=1;i<=m;i++){
    scanf("%s%d%d",ch,&x,&y);
    if(ch[0]=='C')
      q.push_back({x,a[x],-1,0,0}),  //旧数贡献-1
      q.push_back({x,a[x]=y,1,0,0}); //新数贡献+1
    else scanf("%d",&k),q.push_back({x,y,k,i,1});
  }
  memset(ans,-1,sizeof ans);
  solve(q,0,1e9); //整体二分
  for(int i=1;i<=m;i++)if(~ans[i])printf("%d\n",ans[i]);
}
复制代码

 

posted @   董晓  阅读(178)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 单线程的Redis速度为什么快?
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 展开说说关于C#中ORM框架的用法!
点击右上角即可分享
微信分享提示