uva 11997 K Smallest Sums 优先队列处理多路归并问题

题意：K个数组每组K个值，每次从一组中选一个，共K^k种，问前K个小的。

思路：优先队列处理多路归并，每个状态含有K个元素。详见刘汝佳算法指南。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<stack>
#include<queue>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
int n;
int a[1000][1000];
int b[1000];
int c[1000];

struct Item
{
    int s,b;
    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<Item>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();
        c[i]=item.s;
        int B=item.b;
        if(B+1<n)
            q.push(Item(item.s-b[B]+b[B+1],B+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++)
            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;
}