UVa-11997 K Smallest Sums 题解

题意：

给你一个k*k的矩阵，每行选一个数，选出k个数，共有k^k种取法，求这k个数的和。在这k^k个和中，求第1小，第2小……第k小的和并输出。

题解：

首先考虑从这个k*k矩阵的前m行(m>=2)中，每行取一个数，m个数的和中最小的k个已经得到，分别为S(1),S(2)……S(k) 【S(i)代表第i小的和】

则足以得到前m+1行，每行取一个数，m+1个数的和中最小的k个分别是多少。证明如下：

设第m+1行数据分别是A[1],A[2]……A[k]

则由S(1).....S(k)和A[1],A[2]……A[k]进行组合得到的k^2个m+1个数的和一定包含最小的k个

否则 假如S(x)+A[j]【x>k】是前最小的k个和之一，则它之前已经有S(1)+A[j]、S(2)+A[j]……S(k)+A[j]个数比它小了，与它是前k个最小的数矛盾，得证。

所以得到S(1).....S(k)就可以把它组合成一行，然后求它和A[1]......A[k]这一行组成的2*k的矩阵的该问题(即每行取一个数，2个数的和中最小的k个就是前m行中最小的k个(m个数的和))

所以该问题可以降为2*k的等价问题。

设为

a1 a2 a3 ... ak

b1 b2 b3 ... bk

则将b排序后，必有

a1+b1 <= a1+b2 <= a1+b3 ... <= a1+bk

a2+b1 <= a2+b2 <= a2+b3 ... <= a2+bk

......

ak+b1 <= ak+b2 <= ak+b3 ... <= ak+bk

容易知道

S(1)一定在红色区域内【因为a并未排序，所以不是a1+b1】

假如是a2+b1 则它右边的a2+b2和之前的所有待选项中最小的就是S(2)

S(3)...S(k)同理类似。

找到一个最小的 然后把右边的加入候选项 循环这一步。

候选项用一个结构体保存：

struct NODE
{
int s,b;
NODE(){}
NODE(int s,int b) :s(s),b(b) {}
bool operator < (const NODE &other) const
{
return s > other.s;
}
};

注意 重载<为>是因为priority_queue默认大数优先级高。

s为某个a(i)+b(j)的和 b是该和中b的下标j 

之所以只保存b的下标 是因为利用这个就可以很简单的获得其右边的值为 S' = s - b(j) + b(j+1) 

转换为代码 即为

int id = top.b;
		if(id+1<k)
		{
			q.push(NODE(top.s-b[id]+b[id+1],id+1));
		}

完整代码:

#include <cstdio>
#include <queue>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1234;
struct NODE
{
	int s,b;
	NODE(){}
	NODE(int s,int b) :s(s),b(b) {}
	bool operator < (const NODE &other) const
	{
		return s > other.s;
	}
};

int k,a[maxn],b[maxn];
void merge()
{
	priority_queue<NODE> q;
	for(int i=0;i<k;i++)
	{
		q.push(NODE(a[i]+b[0],0));
	}
	for(int i=0;i<k;i++)
	{
		NODE top = q.top(); q.pop();
		a[i] = top.s;
		int id = top.b;
		if(id+1<k)
		{
			q.push(NODE(top.s-b[id]+b[id+1],id+1));
		}
	}
}

int main()
{
	while(~scanf("%d",&k))
	{
		for(int i=0;i<k;i++)
			scanf("%d",&a[i]);
		for(int i=1;i<k;i++)
		{
			for(int j=0;j<k;j++)
				scanf("%d",&b[j]);
			sort(b,b+k);
			merge();
		}
		for(int i=0;i<k;i++)
		{
			printf("%d%c",a[i],i==k-1?'\n':' ');
		}
	}
	return 0;
}