uva 11997 - K Smallest Sums

最新推荐文章于 2019-04-25 21:34:09 发布

二十一画生

最新推荐文章于 2019-04-25 21:34:09 发布

阅读量728

点赞数 1

CC 4.0 BY-SA版权

分类专栏：数据结构简单题文章标签： structure ACM-ICPC uva

本文链接：https://blog.youkuaiyun.com/qhc_20130731/article/details/14253471

简单题同时被 2 个专栏收录

26 篇文章

订阅专栏

数据结构

14 篇文章

订阅专栏

本文介绍了一种高效算法，用于解决从k个整数数组中各取一个元素相加，找出所有可能组合中最小的k个和的问题。通过使用优先队列优化搜索过程，确保了算法效率。

题意：有k个整数数组，各包含k个元素。在每个数组中取一个元素加起来，可以得到^{k^k}个和。求这些和中最小的k个值（从小到大输出）。

#include<cstdio>
#include<algorithm>
#include<queue>

using namespace std;

struct node
{
	int s,b;
	node(int s,int b):s(s),b(b){}
};
bool operator < (node x,node y)
{
	return x.s>y.s;
}
void f(int *A,int *B,int *C,int n)
{
	int i,b;
	priority_queue<node>q;
	for(i=0;i<n;i++)
	{
		q.push(node(A[i]+B[0],0));
	}
	for(i=0;i<n;i++)
	{
		node p=q.top();q.pop();
		C[i]=p.s;
		b=p.b;
		if(b+1<n) q.push(node(p.s-B[b]+B[b+1],b+1));
	}
}
int main()
{
	int i,j,k,A[760],B[760];
	while(scanf("%d",&k)!=EOF)
	{
		for(i=0;i<k;i++) scanf("%d",&A[i]);
		sort(A,A+k);
		for(i=1;i<k;i++) 
		{
			for(j=0;j<k;j++) scanf("%d",&B[j]);
			sort(B,B+k);
			f(A,B,A,k);
		}
		printf("%d",A[0]);
		for(i=1;i<k;i++)
			printf(" %d",A[i]);
		printf("\n");
	}
	return 0;
}