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[][];
 int b[];
 int c[];

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

 int main()
 {
     while(~scanf("%d",&n))
     {
         for(int i=; i<n; i++)
         {
             for(int j=; j<n; j++)
                 scanf("%d",&a[i][j]);
             sort(a[i],a[i]+n);
         }
         for(int i=; i<n; i++)
             merge(a[],a[i],a[],n);
         printf("%d",a[][]);
         for(int i=; i<n; i++)
             printf(" %d",a[][i]);
         printf("\n");
     }
     return ;
 }