UVA 11997 - K Smallest Sums(多路归并)

题目链接:点击打开链接

思路:多路归并。 用贪心的规则, 我们先考虑两个数组的情况, 如果将两个数组从小到大排序, 那么我们先将数组a的所有元素加上数组b的最小的元素, 然后放入优先队列。 因为是事先排好序的, 所以其中肯定有一个是最小的,那么当这个数取出来之后, 如何保证队列头上时次小的呢? 我们发现, a数组的所有元素都用上了, 而且是和b数组中最小的数相加得到的, 所以, 当a[0]+b[0]被取出, 下一个可能有用的和一定不是a[1]+b[0], 因为已经在队列中了, 那么只能是a[0] + b[1]。 贪心的每次取出一个数之后再加进去一个尽量小的值。

本题进行k次就行了。

细节参见代码:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 1000;
int T,n,m,a[maxn][maxn];
struct node {
    int s, b;
    node(int s=0, int b=0):s(s), b(b) {}
    bool operator < (const node& rhs) const {
        return s > rhs.s;
    }
};
void bin(int* a, int *b, int *c, int n) {
    priority_queue q;
    for(int i = 0; i < n; i++) q.push(node(a[i]+b[0], 0));
    for(int i = 0; i < n; i++) {
        node cur = q.top(); q.pop();
        c[i] = cur.s;
        int id = cur.b;
        if(id+1 < n) q.push(node(cur.s-b[id]+b[id+1], id+1));
    }
}
int main() {
    while(~scanf("%d",&n)) {
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) scanf("%d",&a[i][j]);
            sort(a[i], a[i]+n);
        }
        for(int i = 1; i < n; i++) {
            bin(a[0], a[i], a[0], n);
        }
        for(int i = 0; i < n; i++) {
            printf("%d%c", a[0][i], i == n-1 ? '\n' : ' ');
        }
    }
    return 0;
}



你可能感兴趣的:(uva解题报告,数据结构,ACM竞赛)