静态主席树求区间第k小

静态主席树求区间第k小

Problem:
给定一个含有 n 个数的序列 a1,a2…an,需要支持两种操作:

Q l r k 表示查询下标在区间 [l,r] 中的第 k 小的数。
C x y 表示将 ax 改为 y。

/*
给定一个含有 n 个数的序列 a1,a2…an,需要支持两种操作:

Q l r k 表示查询下标在区间 [l,r] 中的第 k 小的数。
C x y 表示将 ax 改为 y。
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=200005;

inline ll read() {
	ll x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9') {
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
	return x*f;
}
//***************************************
ll n,m,T,tot=0,x,y,k;
ll a[maxn];
ll root[maxn];
struct node {
	ll l,r,sum;
} t[maxn<<5];
vector<ll>v;
//***************************************
void discreat() {//离散化
	v.clear();
	for (int i=1; i<=n; i++) v.push_back(a[i]);
	sort(v.begin(), v.end());
	v.erase(unique(v.begin(), v.end()), v.end());
}
inline ll getid(ll x) {
	return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
void update(ll pre,ll &now,ll l,ll r,ll p) {
	t[++tot]=t[pre];
	now=tot;
	t[now].sum++;
	if(l==r)return ;
	ll mid=(l+r)>>1;
	if(p<=mid) update(t[pre].l,t[now].l,l,mid,p);
	else update(t[pre].r,t[now].r,mid+1,r,p);
}
ll query(ll ql,ll qr,ll l,ll r,ll k) {
	if(l==r) return l;
	ll tmp=t[t[qr].l].sum-t[t[ql].l].sum;
	ll mid=(l+r)>>1;
	if(k<=tmp) return query(t[ql].l,t[qr].l,l,mid,k);
	else return query(t[ql].r,t[qr].r,mid+1,r,k-tmp);
}
int main() {
//	scanf("%lld",&T);
//	while(T--){
	n=read(),m=read();
//		scanf("%lld%lld",&n,&m);

//	tot=0;
//	memset(a,0,sizeof(a));
//	memset(root,0,sizeof(root));

//	for(int i=1; i<=n; i++) scanf("%lld",&a[i]);
	for(int i=1; i<=n; i++)a[i]=read();
	discreat();

	for(int i=1; i<=n; i++) {
		update(root[i-1],root[i],1,n,getid(a[i]));
	}

	while(m--) {
//		scanf("%lld%lld%lld",&x,&y,&k);
		x=read(),y=read(),k=read();
		ll pos=query(root[x-1],root[y],1,n,k);
		printf("%lld\n",v[pos-1]);
	}
//	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值