每次插入一个数字,在第u[i]次插入完后可以查询第k大的数字,然后k递增,初始化k=1。
总共数字n<30000,查询m<=n。
分析:很容易想到用优先队列去维护已插入的序列有序,然后难处理的就是怎么找到第K大的数字?因为k是递增的,所以每次查询完以后,前k小就不会再出现了,但是可能以后会插入比前k小中某个数还小的,这时候就要想办法把当前插入的数和第k小交换一下。所以可以再利用一个大根堆的优先队列,去保存已经查询过的K个数,然后新插入的数再跟最大数比较,判断是否交换处理即可。
#include<cstdio> #include<cstring> #include<algorithm> #include<queue> #include<vector> using namespace std; const int N=30009; int a[N],u[N],m,n; int main() { scanf("%d%d",&m,&n); for(int i=0;i<m;i++)scanf("%d",&a[i]); for(int i=0;i<n;i++)scanf("%d",&u[i]); priority_queue<int>maxq; //大根堆,保存已查询的前k小的数 priority_queue<int,vector<int>,greater<int> >minq; //小根堆,队首元素是第k+1小 int t=0; for(int i=0;i<m;i++){ minq.push(a[i]); while(!maxq.empty()&&maxq.top()>minq.top()){ int maxt=maxq.top();maxq.pop(); int mint=minq.top();minq.pop(); maxq.push(mint); minq.push(maxt); } while(t<n&&u[t]==i+1){ printf("%d ",minq.top()); maxq.push(minq.top()); minq.pop(); t++; } } return 0; }