Uva11997 K Smallest Sums（优先队列）

题意： 有一个k*k 的方阵，让你从当中每一行挑选一个数字相加最后能得到K^K次方的和，输出其中最小的k个。

分析：《训练指南》P190，先对每一行排序然后两两归并，每次取前k个再和下一行再进行归并。

代码：

#include
using namespace std;

struct Item {
  int s, b; // s=A[a]+B[b]。这里的a并不重要，因此不保存
  Item(int s, int b):s(s), b(b) { }
  bool operator < (const Item& rhs) const {
    return s > rhs.s;
  }
};

void merge(int* A, int* B, int* C, int n) {
  priority_queue q;
  for(int i = 0; i < n; i++)
    q.push(Item(A[i]+B[0], 0));
  for(int i = 0; i < n; i++) {
    Item item = q.top(); q.pop();// 取出A[a]+B[b]
    C[i] = item.s;
    int b = item.b;
    if(b+1 < n) q.push(Item(item.s-B[b]+B[b+1], b+1)); // 加入A[a]+B[b+1]=s-B[b]+B[b+1]
  }
}

const int maxn = 768;
int A[maxn][maxn];

int main() {
  int n;
  while(scanf("%d", &n) == 1) {
    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++) // 两两合并
      merge(A[0], A[i], A[0], n); // (*)

    printf("%d", A[0][0]); // 输出结果
    for(int i = 1; i < n; i++)
      printf(" %d", A[0][i]);
    printf("\n");
  }
  return 0;
}