树状数组+二分查询第K大(经典应用)

题目链接

按题意每一次查询和最大的p个数的最小字典序中第q个位置的数字；
首先我们可以把所有数字的按从大到小进行一级排序，位置按从前往后进行二级排序，那么在排序以后p个数的最大和且最小字典序就是前p个数，那么我只要输出前面按照位置从小到大的第q个数就行了。
显然，问题转化后就是求前p个数中的第q大值；
树状数组维护数字x是第几大，然后在有单调性的情况下选择二分第q大值就好了。

#include<stdio.h>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<string.h>
using namespace std;
const int N = 2e5+5;
typedef long long ll;
int a[N],ans[N],n,sum[N];
struct ab{
	int x,y;
}c[N];
struct sc{
	int x,y,z;
}b[N];
bool cmp(ab p,ab q){
	if(p.x>q.x) return true;
	if(p.x==q.x&&p.y<q.y) return true;
	return false;
}
bool cmp1(sc p,sc q){
	return p.x<q.x;
}
void add(int x){
	while(x<=n){
		//printf("%dxxx\n",x);
		sum[x]++;
		x+=(x&(-x));
	}
}
int get(int x){
	int ans=0;
	while(x){
		ans+=sum[x];
		x-=(x&(-x));
	}
	return ans;
}
int main()
{
	int t,m,i,j;
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&a[i]);
		c[i].x=a[i];
		c[i].y=i;
	}
	sort(c+1,c+1+n,cmp);
	for(i=1;i<=n;i++) sum[i]=0;
	scanf("%d",&m);
	for(i=1;i<=m;i++){
		scanf("%d %d",&b[i].x,&b[i].y);
		b[i].z=i;
	}
	sort(b+1,b+1+m,cmp1);
	int p=b[m].x,q=1;
	for(i=1;i<=p;i++){
		add(c[i].y);
		//printf("%dss\n",c[i].y);
		while(b[q].x==i){
			int l=1,r=n,mid=0;
			while(l<r){
				mid=(l+r)/2;
				//printf("%d %d\n",mid,get(mid));
				if(get(mid)>=b[q].y) r=mid;
				else l=mid+1;
			}
			ans[b[q].z]=a[r];
			q++;
		}
	}
	for(i=1;i<=m;i++){
		printf("%d\n",ans[i]);
	}
	return 0;
}