主席树模板求助
查看原帖
主席树模板求助
549999
sail_with_pleasure楼主2024/9/19 15:32

本地第一组数据过了,但是洛谷上全WA。

应该是某种特殊错误QwQ

#include<bits/stdc++.h>
using namespace std;
const int N=7e6,z=1e6;
int n,m,a[z],top,cnt,root[z],num[N],book[N],lson[N],rson[N],ll,rr,k,sum;
set<int> qq;
map<int,int> q;
void crt(int s,int l,int r){
	int mid=(l+r)/2;
	if(l==r)return;
	lson[s]=++top;
	rson[s]=++top;
	crt(lson[s],l,mid);
	crt(rson[s],mid+1,r);
	return;
}
void cnew(int s,int l,int r,int x){
	int mid=(l+r)/2;
	if(l==r){
		num[++top]=num[s]+1;
		return;
	}
	if(x<=mid){
		cnew(lson[s],l,mid,x);
		top++;
		lson[top]=top-1;
		rson[top]=rson[s];
	}else{
		cnew(rson[s],mid+1,r,x);
		top++;
		rson[top]=top-1;
		lson[top]=lson[s];
	}
	num[top]=num[lson[top]]+num[rson[top]];
	return;
}
int ans(int sl,int sr,int l,int r){
	int mid=(l+r)/2;
	if(l==r){
		return l;
	}
	if(k>num[lson[sr]]-num[lson[sl]]+sum){
		sum+=num[lson[sr]]-num[lson[sl]];
		return ans(rson[sl],rson[sr],mid+1,r);
	}else{
		return ans(lson[sl],lson[sr],l,mid);
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		qq.insert(a[i]);
	} 
	for(set<int>::iterator it=qq.begin();it!=qq.end();it++){
		q[*it]=++cnt;
		book[cnt]=*it;
	}
	root[0]=1;
	crt(++top,1,cnt);
	for(int i=1;i<=n;i++){
		a[i]=q[a[i]];
		cnew(root[i-1],1,n,a[i]);
		root[i]=top;
	}
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&ll,&rr,&k);
		sum=0;
		printf("%d\n",book[ans(root[ll-1],root[rr],1,n)]);
	}
	return 0;
}
2024/9/19 15:32
加载中...