uva 11997 （基础数据结构）

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

思路：先对每一行排序然后两两归并，每次取前k个再和下一行再进行归并。在归并的时候用一个优先队列维护最大的k个值每次先放k个进去然后一次每行和队顶比较，若是小则替换否则break最后输出即可。

代码如下：

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <set>
#include <string.h>
#include <map>
#include <vector>
#include <string>
#include <math.h>
using namespace std;

const int LEN = 2010;
int Num[LEN][LEN];
int s[LEN], n;

void Merge(int a, int b){
    int top = 0;
    priority_queue<int> q;
    for(int i=0; i<n; i++){
        q.push(Num[a][i] + Num[b][0]);
    }
    for(int i=0; i<n; i++){
        for(int j=1; j<n; j++){
            int ls = Num[a][i] + Num[b][j];
            int nv = q.top(); q.pop();
            if(nv > ls){ 
                swap(nv, ls);
                q.push(nv);
            }else {
                q.push(nv);
                break;
            }
        }
    }
    while(!q.empty()){
        s[top++] = q.top();q.pop();
    }
    for(int i=top-1, j=0; i>=0; i--, j++){
        Num[b][j] = s[i];
    }
}

int main()
{
//    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);

    ios::sync_with_stdio(false);
    while(cin >> n){
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                cin >> Num[i][j];
            }
            sort(Num[i], Num[i]+n);
        }
        for(int i=1; i<n; i++){
            Merge(i-1, i);
        }
        for(int i=n-1; i>=0; i--){
            cout << s[i];
            if(i != 0) cout << ' ';
        }cout << endl;
    }
    return 0;
}