POJ 1442 Black Box

题目链接:Black Box

解题思路:这里可以用两个堆来实现,一个最大堆一个最小堆,最大堆可以用来保存i之前的数字序列,最小堆可以放i以及之后的数字序列,保证最大堆的数字都比最小堆的数字小。每一次出队的的数字就是get的答案,再进入最大堆中。

#include<stdio.h>
#include<queue>
#include<vector>
#include<algorithm>
#define MAX 30010

using namespace std;

int num[MAX], a[MAX]; 

int main(){
	int i, j, n, m, tem;
	//freopen("in.txt", "r", stdin);
	while(scanf("%d%d", &n, &m) != EOF){
		priority_queue<int, vector<int>, greater<int> > Squeue;
		priority_queue<int, vector<int>, less<int> > Gqueue;
		memset(a, 0, sizeof(a));
		for(i = 1; i <= n; i++){
			scanf("%d", &num[i]);
		}
		for(i = 1; i <= m; i++){
			scanf("%d", &a[i]);
		}
		j = 1;
		for(i = 1; i <= n; i++){
			Squeue.push(num[i]);
			if(!Gqueue.empty() && Squeue.top() < Gqueue.top()){
				int S = Squeue.top();
				int G = Gqueue.top();
				Squeue.pop();
				Gqueue.pop();
				Squeue.push(G);
				Gqueue.push(S);
			}
			while(a[j] == i){
				printf("%d\n", Squeue.top());
				Gqueue.push(Squeue.top());
				Squeue.pop();
				j++;
			}
		}
	}
	return 0;
}


 

你可能感兴趣的:(POJ 1442 Black Box)