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;
}