UVa - 11997 - K Smallest Sums

这个题目的意思就是给出一个 k*k 的二维数组   从每一行取一个数相加 然后一共可以得到k^k种结果 现在要求这些结果中最小的k个 按升序打印出来。

一开始的想法太简单了 wa掉了 想了一下知道为什么wa了

之前的思路是 找出k个数组中每个数组的最小值 这样得到的肯定是最小的

那么再找其余k-1个最小值 我保存了 每个数组中的值减去那个最小值得到的结果

这样就想 每次换一个数 尽量换小的。

其实可以两个数组都换 比如第一个数组 有一个大一的 第二个也有一个大一的 剩下所有的元素都大三以上

这样一二数组都换大一的 才小。。。

这就是错误的原因

接下来就按多路归并做了

我们先想简化版的

有两个数组A和B size都是k 现在要从A和B中各选一个元素 取和 找出最小的k个和

就假设是这个样两个数组吧

1 8 5

9 2 5

我们先排序

1 5 8

2 5 9

这样最小的就很明显了 肯定是3

定义一个结构体

struct Node

{   int sum,b;

};

构造函数和重载运算符就不写了

我们找最小值其实是通过优先队列来找的

因为A和B已经排序好了  

B[0]肯定是B中最小值 我们往优先队列里push  k个数 分别是a[0]+b[0] 到 a[k-1]+b[0];

然后用优先队列弹出来k次 就得到了最小值

每次弹出来一个 加入是 node  那么就要加进去下一个 下一个的sum就要变成原来的减去那个b加下一个B

这样得到了两个数组的最小的k个数

那么k个数组呢 一个道理 先找出前两个数组的最小的k个数字

再用这个结果和第三个数组进行归并。。

最终得到答案了

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;

const int maxk=751;
int ans[maxk],arr[maxk],k;

struct Node
{
    int sum,b;
    Node(int i,int j):sum(i),b(j){}
    bool operator < (const Node & n)const
    {
        return sum>n.sum;
    }
};

void merge()
{
    priority_queue<Node> pq;
    for(int i=0;i<k;++i)
    pq.push(Node(ans[i]+arr[0],0));

    for(int i=0;i<k;++i)
    {
        Node node=pq.top();
        pq.pop();
        ans[i]=node.sum;
        if(node.b+1<k)
        {
            node.sum=node.sum-arr[node.b]+arr[node.b+1];
            node.b++;
            pq.push(node);
        }
    }
}

int main()
{

    int i,j;
    while(scanf("%d",&k)!=EOF)
    {
        for(i=0;i<k;++i)
        scanf("%d",&ans[i]);
        sort(ans,ans+k);
        for(i=1;i<k;++i)
        {
            for(j=0;j<k;++j)
                scanf("%d",&arr[j]);
            sort(arr,arr+k);
            merge();
        }
        printf("%d",ans[0]);
        for(i=1;i<k;++i)
        printf(" %d",ans[i]);
        printf("\n");
    }
    return 0;
}