动态第K大 线段树套权值线段树 / 树状数组套权值线段树

下面两个程序已经对拍  放心食用

树状数组套权值线段树

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=2e6+5;
int rt[maxn*2],ls[maxn*20],rs[maxn*20],sum[maxn*20];
int n,tot,sz,cnt,a[maxn],b[maxn*2],c[maxn];
int V1[maxn],V2[maxn],c1,c2;
struct node
{
	int l,r,k,flag;
}q[maxn];
void update(int &o,int l,int r,int k,int value)
{
	if(!o)o=++cnt;
	sum[o]+=value;
	if(l==r)return;
	int m=(l+r)/2;
	if(k<=m)update(ls[o],l,m,k,value);
	else update(rs[o],m+1,r,k,value); 
}
void modify(int pos,int value)
{
	int k=lower_bound(b+1,b+1+sz,a[pos])-b;
	for(int i=pos;i<=n;i+=(i&-i))
	update(rt[i],1,sz,k,value);
}
int query(int l,int r,int k)
{
	if(l==r)return l;
	int m=(l+r)/2,res=0;
	for(int i=1;i<=c1;i++)res+=sum[ls[V1[i]]];
	for(int i=1;i<=c2;i++)res-=sum[ls[V2[i]]];
	if(k<=res)
	{
		for(int i=1;i<=c1;i++)V1[i]=ls[V1[i]];
		for(int i=1;i<=c2;i++)V2[i]=ls[V2[i]];
		return query(l,m,k);
	}
	else
	{
		for(int i=1;i<=c1;i++)V1[i]=rs[V1[i]];
		for(int i=1;i<=c2;i++)V2[i]=rs[V2[i]];
		return query(m+1,r,k-res);		
	}
}
int get(int l,int r,int k)
{
	c1=c2=0;
	for(int i=r;i;i-=(i&-i))
	V1[++c1]=rt[i];
	for(int i=l-1;i;i-=(i&-i))
	V2[++c2]=rt[i];
	return query(1,sz,k);
}
int main()
{
	int m,l,r,k;
	char s[2];
	tot=0;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]),b[++tot]=a[i];
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s);
		if(s[0]=='C')
		{
			scanf("%d%d",&q[i].l,&q[i].k);
			q[i].flag=1,b[++tot]=q[i].k;
		}
		else
		{
			scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
			q[i].flag=2;
		}	
	}
	sort(b+1,b+1+tot);
	sz=unique(b+1,b+1+tot)-(b+1);
	for(int i=1;i<=n;i++)
	modify(i,1);
	for(int i=1;i<=m;i++)
	{
		if(q[i].flag==1)
		{
			modify(q[i].l,-1);
			a[q[i].l]=q[i].k;
			modify(q[i].l,1);
		}
		else
		printf("%d\n",b[get(q[i].l,q[i].r,q[i].k)]);
	}
}

 

这种写法是线段树套权值线段树

#include<bits/stdc++.h>
typedef long long ll;
#define ls1 o*2
#define rs1 o*2+1
#define pb push_back
const int M =2500000+7;
const int N =6e4;
using namespace std;
int a[N],b[N];
int ls[M],rs[M],rt[M],sm[M],ct,sz,n;
struct node
{
	int l,r,k,flag;
}q[N];
void up(int &o,int l,int r,int x,int d)
{
	if(!o)o=++ct;
	sm[o]+=d;
	if(l==r)return ;
	int m=(l+r)/2;
	if(x<=m)up(ls[o],l,m,x,d);
	else up(rs[o],m+1,r,x,d);
}
void modify(int o,int l,int r,int x,int d,int t)
{
	up(rt[o],1,sz,d,t);
	if(l==r)return;
	int m=(l+r)/2;
	if(x<=m)modify(ls1,l,m,x,d,t);
	else modify(rs1,m+1,r,x,d,t);
}
vector<int>v;
void gt(int o,int l,int r,int x,int y)
{
	if(x<=l&&r<=y)
	{
		v.pb(rt[o]);
		return ;
	}
	int m=(l+r)/2;
	if(x<=m)gt(ls1,l,m,x,y);
	if(y>m)gt(rs1,m+1,r,x,y);
}
int qu(int l,int r,int k)
{
	int ans=0;
	for(auto x:v)ans+=sm[ls[x]];
	if(l==r)return l;
	int m=(l+r)/2;
//	cout<<l<<"  "<<r<<"  "<<k<<"  "<<ans<<endl;
	if(ans>=k)
	{
		for(int i=0;i<v.size();i++)v[i]=ls[v[i]];
		return qu(l,m,k);
	}
	for(int i=0;i<v.size();i++)v[i]=rs[v[i]];
	return qu(m+1,r,k-ans);
}
int get(int l,int r,int k)
{
	v.clear();
	gt(1,1,n,l,r);
	return qu(1,sz,k);
}
int main()
{
//	freopen("1.in","r",stdin);
//	freopen("1.ans","w",stdout);
	int m,l,r,k;
	char s[2];
	ct=0;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]),b[++sz]=a[i];
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s);
		if(s[0]=='C')
		{
			scanf("%d%d",&q[i].l,&q[i].k);
			q[i].flag=1,b[++sz]=q[i].k;
		}
		else
		{
			scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
			q[i].flag=2;
		}	
	}
	sort(b+1,b+1+sz);
	sz=unique(b+1,b+1+sz)-(b+1);
	for(int i=1;i<=n;i++)a[i]=lower_bound(b+1,b+1+sz,a[i])-b;
//	cout<<i<<"  "<<a[i]<<endl;;
	for(int i=1;i<=n;i++)
		modify(1,1,n,i,a[i],1);
	for(int i=1;i<=m;i++)
	{
		if(q[i].flag==1)
		{
			modify(1,1,n,q[i].l,a[q[i].l],-1);
			a[q[i].l]=lower_bound(b+1,b+1+sz,q[i].k)-b;
			modify(1,1,n,q[i].l,a[q[i].l],1);
		}
		else
		printf("%d\n",b[get(q[i].l,q[i].r,q[i].k)]);
	}
	return 0;	
}
/*
2 
5 1
3 2 1 4 7
Q 1 4 3
*/