17AHU排位赛3 F题 (多路归并求第k大和)

problem

蕊蕊有n部喜欢的番剧，每部番剧里面有m个老婆，每个老婆都有一个颜值。现在蕊蕊想从每部番剧里挑出一个老婆，总共挑出n个老婆，组成后宫团，后宫团的颜值是里面n个老婆的颜值和。显然，蕊蕊一共可以组成m^n个后宫团，现在她想知道这m^n个后宫团中，颜值和第K大的总颜值是多少。若K>m^n，输出0。

Input

第一行三个整数n,m,k(1<=n<=1000,1<=m<=500,1<=k<=1000)。
接下来n行，每一行表示一部番剧的信息。每行m个整数，表示该番剧里m个老婆的颜值a(1<=a<=100000)。

Output

输出一个整数，表示颜值和第K大的后宫团总颜值。

Input

2 3 4
1 4 3
2 2 5

Output

6

Limitation

1s 256MB

Hint

蕊蕊一共可以组成3^2=9个后宫团，总颜值分别是：
1+2=3,1+2=3,1+5=6
4+2=6,4+2=6,4+5=9
3+2=5,3+2=5,3+5=8
其中第4大的是6

思路

这题是一个多路归并问题，一般求第k大和。

思路就是准备一个ans[]数组 用来存放前k大和
然后进行n次归并，每一次归并将ans数组和a数组（当前m个选择）进行组合（相加）放进优先队列（或堆）里，维护最大值，次大值……然后更新ans[]数组。

---

到最后一次归并，将前n-1次归并的ans数组 组合加上最后的a数组并维护前k大，ans[k]即为所求。如果所有的组合可能数都没有到k,那ans[k]为0。

时间复杂度：O(n * k* log(m) )

代码示例

//多路归并,优先队列实现
#include<bits/stdc++.h>
using namespace std;

const int maxn=1005;

int a[maxn],ans[maxn],w[maxn];

int n,m,k;

bool cmp(const int x,const int y){
    //数组从大到小排序
    return x>y;
}

struct cyx{
    int x,y,v;
    //重载运算符，小于号时比的是v,不重载会出错，因为是优先队列的元素
    bool operator < (const cyx & p) const {
        return v<p.v;
    }
};

priority_queue<cyx> q;//优先队列，基本元素是cyx

void Merge()
{
    if(ans[0]==0){//第一次归并
        ans[0]=a[0];//ans[0]等于每个后宫团美眉的数量(m)
        for(int i=1;i<=ans[0];++i) ans[i]=a[i];
        //第一次嘛，先存第一个后宫团的排序值(由大到小)
        return ;
    }
    while(!q.empty()) q.pop();//清空优先队列
    int n=ans[0],m=a[0];
    w[0]=0;//w[0]计数,表示当前的前k大,k<=1000
    for(int i=1;i<=n;++i) q.push((cyx){i,1,ans[i]+a[1]});
    for(int i=0;i<k;++i){
        if(q.empty()) break;
        cyx u=q.top();
        q.pop();
        w[++w[0]]=u.v;//记录一下
        if(u.y<m) q.push((cyx){u.x,u.y+1,ans[u.x]+a[u.y+1]});
    }
    ans[0]=w[0];
    for(int i=1;i<=ans[0];++i) ans[i]=w[i];
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    while(n--)//n个后宫团，归并n次
    {
        for(int i=1;i<=m;++i) cin>>a[i];//每一个后宫团有m个美眉
        a[0]=m;//a[0]记录数量
        sort(a+1,a+m+1,cmp);//对当前后宫团的颜值进行排序
        Merge();
    }
    //for(int i=1;i<=ans[0];++i) cout<<ans[i]<<endl;
    cout<<ans[k]<<endl;//输出总第k大后宫团组合
    return 0;
}