洛谷P1631 序列合并

题意：有分别含N个元素的两个数组，两个数组每个元素相加有N*N个和，求这里面的N个最小值
思路：两数组已排好序，其中a中第一个元素必须选，故将b中所有元素与其相加入堆，取p为b中k对应元素与a中元素取和的a的索引，每次取出一个元素后将该索引递增并相加入堆，就不会出现重复问题

#include<bits/stdc++.h>
using namespace std;
int n, a[100005], b[100005], p[100005];
int main()
{
	while (cin >> n) {
		priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > q;
		for (int i = 0; i < n; i++) {
			cin >> a[i];
		}
		for (int i = 0; i < n; i++) {
			cin >> b[i];
			q.push(make_pair(a[0]+b[i], i));
			p[i] = 0;
		}
		for (int i = 0; i < n; i++) {
			printf("%d ", q.top().first);
			int k = q.top().second;
			q.pop();
			q.push(make_pair(a[++p[k]]+b[k], k));
		}
	}
	return 0;
}