求助主席树(没过样例)
查看原帖
求助主席树(没过样例)
658786
STUDENT00楼主2022/11/20 10:24

萌新刚学主席树,便来水模板题,然而发现自己连样例都过不了。。。

#include<bits/stdc++.h>
#define N 200010
using namespace std;
int n,m,a[N],b[N],q,rt[N<<5],lc[N<<5],rc[N<<5],sum[N<<5],cnt;
void build(int t,int l,int r){
	if(l==r) return;
	int mid=l+r>>1;
	lc[t]=++cnt;
	build(lc[t],l,mid);
	rc[t]=++cnt;
	build(rc[t],mid+1,r);
}
int update(int l,int r,int fa,int k){
	int rt=++cnt;
	lc[rt]=lc[fa];
	rc[rt]=rc[fa];
	sum[rt]=sum[fa]+1;
	if(l==r) return rt;
	int mid=l+r>>1;
	if(k<=mid) lc[rt]=update(l,mid,lc[rt],k);
	else lc[rt]=update(mid+1,r,rc[rt],k);
	return rt;
}
int query(int l,int r,int a,int b,int k){
	int mid=l+r>>1,x=sum[lc[b]]-sum[lc[a]];
	if(l==r) return l;
	if(x>=k) return query(l,mid,lc[a],lc[b],k);
	else return query(mid+1,r,rc[a],rc[b],k-x);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	memcpy(b,a,sizeof(b));
	sort(b+1,b+n+1);
	q=unique(b+1,b+n+1)-b-1;
	build(rt[0],1,q);
	for(int i=1;i<=n;i++) rt[i]=update(1,q,rt[i-1],lower_bound(b+1,b+q+1,a[i])-b);
	while(m--){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",b[query(1,q,rt[l-1],rt[r],k)]);
	}
	return 0;
}
2022/11/20 10:24
加载中...