题意:有k个组,每组k个数,要从每组挑出一个数,k个组相加,一共有k^k种和,要求升序输出k个最小的和。
题解:书上的题解的方法是用优先队列和两两归并,先看两个排好序的数组,(s, b)代表一个元素和,让s = Aa+Bb,那么下一个可能元素和就是(s', b + 1),s' = s - Bb + Bb+1,然后优先队列每次取出队首就是此时队伍中和最小的情况,然后得到s'入队,因为是排好序就不会重复,循环k - 1次得到k个数组中最小的k个和。
#include <stdio.h>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 755;
int n, arr[N][N];
struct Item {
int s, b;
Item(int s, int b): s(s), b(b) {}
bool operator < (const Item& temp) const {
return s > temp.s;
}
};
void merge(int* A, int* B, int* C) {
priority_queue<Item> q;
for (int i = 0; i < n; i++)
q.push(Item(A[i] + B[0], 0));
for (int i = 0; i < n; i++) {
int bb = q.top().b;
C[i] = q.top().s;
q.pop();
if (bb + 1 < n)
q.push(Item(C[i] - B[bb] + B[bb + 1], bb + 1));
}
}
int main() {
while (scanf("%d", &n) == 1) {
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++)
scanf("%d", &arr[i][j]);
sort(arr[i], arr[i] + n);
}
for (int i = 1; i < n; i++)
merge(arr[0], arr[i], arr[0]);
printf("%d", arr[0][0]);
for (int i = 1; i < n; i++)
printf(" %d", arr[0][i]);
printf("\n");
}
return 0;
}