UVA11997 K Smallest Sums(思维题/优先队列)

题目

k(2<=k<=750)个数组，每个数组k个元素，

从每个数组取1个元素，所组成的k元组之和，共有种可能

现在要求，输出其中的前k小值

思路来源

https://www.cnblogs.com/yyf0309/p/7797202.html

题解

两两合并，每两个合并的复杂度O(klogk)，则总复杂度O(k²logk)

对第二个数组排好序后，考虑两个的合并过程，

优先队列内放入数对(第一个的值，第二个的当前位置0)

那么，第二个数组内位于pos+1的元素被取，

当且仅当在固定第一个数组的值的情况下，第二个数组内pos位置的元素被取

所以，就类似k短路的过程，优先取近的，再把一步可达的加进优先队列，

优先队列，自然是按对答案的贡献重载运算符

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=755;
int k;
int a[maxn],b[maxn];
struct node
{
	int v,pos;
	node(){}
	node(int vv,int p):v(vv),pos(p){}
}t;
bool operator<(node c,node d)
{
	return c.v+b[c.pos]>d.v+b[d.pos];
}
priority_queue<node>q;
void unite()
{
	while(!q.empty())q.pop();
	for(int j=0;j<k;++j)
	q.push(node(a[j],0));//a的数值 b的下标 
	int cnt=0;
	while(!q.empty()&&cnt<k)
	{
		t=q.top();q.pop();
		//b只有这个下标被取了 下一个才有被取的可能 
		a[cnt++]=t.v+b[t.pos];
		if(t.pos+1<k)q.push(node(t.v,t.pos+1));
	}
}
int main()
{
	while(~scanf("%d",&k))
	{
		for(int j=0;j<k;++j)
		scanf("%d",&a[j]);
		sort(a,a+k);
		for(int i=1;i<k;++i)
		{
			for(int j=0;j<k;++j)
			scanf("%d",&b[j]);
			sort(b,b+k);
			unite();
		}
		for(int i=0;i<k;++i)
		printf("%d%c",a[i],i==k-1?'\n':' ');
	}
	return 0;
}