ABC440_D题_题解

题目链接。

把出现过的标记为 1 1 1，没有出现过的就是 0 0 0 了，查询第 Y Y Y 小的直接用线段树二分。

解决要大于等于 X X X 这一限制也很简单，设 1 ∼ X 1\sim X 1∼X 中有 x x x 个数没被标记，大于等于 X X X 中的第 Y Y Y 小，就是所有数中第 Y + x Y+x Y+x 小，线段树板子题。

#include<bits/stdc++.h>
#define ll long long
#define ull usngiend long long

using namespace std;
//谨慎多开几个 long long 
const int N=300005;
const ll V=2100000000;

ll a[N];
struct node{
	int lc,rc;
	ll dat;
}t[5*19*N];
int rt,cnt;
void change(int &p,ll l,ll r,ll x,ll v){
	if(!p)p=++cnt;
	if(l==r){t[p].dat+=v;return;}
	ll mid=(l+r)>>1;
	if(x<=mid)change(t[p].lc,l,mid,x,v);
	else change(t[p].rc,mid+1,r,x,v);
	t[p].dat=t[t[p].lc].dat+t[t[p].rc].dat;
	return;
}
ll ask(int p,ll l,ll r,ll L,ll R){
	if(!p)return 0;
	if(L<=l&&r<=R)return t[p].dat;
	ll mid=(l+r)>>1,sum=0;
	if(L<=mid)sum+=ask(t[p].lc,l,mid,L,R);
	if(R>mid)sum+=ask(t[p].rc,mid+1,r,L,R);
	return sum;
}
ll get(int p,ll l,ll r,ll k){
	if(l==r)return l;
	ll mid=(l+r)>>1;
	ll d=(mid-l+1)-t[t[p].lc].dat;
	if(d>=k)return get(t[p].lc,l,mid,k);
	else return get(t[p].rc,mid+1,r,k-d);
}
int main(){
	int n,q;ll x,y;
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
	for(int i=1;i<=n;i++)change(rt,1,V,a[i],1);
	while(q--){
		scanf("%lld%lld",&x,&y);
		printf("%lld\n",get(rt,1,V,y+x-1-ask(rt,1,V,1,x-1)));
	}
	return 0;
}