uva 11997(priority_queue)

来源：互联网发布：mac命令行安装mysql 编辑：程序博客网时间：2024/05/29 19:54

题意：有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;}

0 0