codeforces1261 B2. Optimal Subsequences (Hard Version)

https://codeforces.com/problemset/problem/1261/B2

先对原数组排序，先按数字大小从大到小，再按下标从小到大。

对于每个询问k，选择每个数字哪几种肯定是确定了，只是顺序不确定而已，那么如果按下标从小到大排，对于相同数字只拿一部分的情况，就能贪心得选择下标较小的那部分。

那么要求第pos个的位置，就是求排了序后前k个下标中第pos小的是谁就行了，很经典的主席树求区间第k小。

直接从模板里复制主席树了，其实也可以离线询问然后线段树搞，强制在线就没法了。

#include<bits/stdc++.h>
#define maxl 200010
using namespace std;

int n,m,ans,tot;
int b[maxl],rt[maxl];
struct nu
{
	int val,id;
}a[maxl];
char s[maxl];
struct node
{
	int ls,rs,sum;
}tree[maxl*20];

inline bool cmp(const nu &x,const nu &y)
{
	if(x.val==y.val)
		return x.id<y.id;
	else
		return x.val>y.val;
}

inline void insert(int num,int &x,int l,int r)
{
	tree[++tot]=tree[x];x=tot;
	++tree[x].sum;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(num<=mid)
		insert(num,tree[x].ls,l,mid);
	else
		insert(num,tree[x].rs,mid+1,r);
}

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i].val),a[i].id=i,b[i]=a[i].val;
	sort(a+1,a+1+n,cmp);
	tot=0;
	for(int i=1;i<=n;i++)
	{
		rt[i]=rt[i-1];
		insert(a[i].id,rt[i],1,n);
	}
}

inline int query(int i,int j,int k,int l,int r)
{
	if(l==r)
		return l;
	int tp=tree[tree[j].ls].sum-tree[tree[i].ls].sum;
	int mid=(l+r)>>1;
	if(k<=tp) 
		return query(tree[i].ls,tree[j].ls,k,l,mid);
	else
		return query(tree[i].rs,tree[j].rs,k-tp,mid+1,r);
}

inline void mainwork()
{
	scanf("%d",&m);
	int k,pos;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&k,&pos);
		ans=query(rt[0],rt[k],pos,1,n);
		printf("%d\n",b[ans]);
	}
}

inline void print()
{
	
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}