多校联合（4） ZZ的橱柜

题目链接：http://www.acmore.net/problem.php?id=1504

Description

ZZ不仅喜欢买衣服，还是个吃货，天哪！现在的妹纸是怎么了？╮(╯_╰)╭，ZZ有两个大型橱柜A和B，里面放满了零食,每个橱柜里面有N个方格，每个方格里面放了不同重量的美味，现在ZZ要从A、B里面各取一个方格的美味，可是ZZ不想变得太胖，于是他会尽量的选择重量少的美味，当然不是所有的组合都是ZZ喜欢吃的美味，所以她想了一个特别的方法，他会选出组合重量最小的前M种中选出一种，可是怎么才能算出前M个组合的重量的呢？这个真是个难题，作为ACMer的ZZ竟然不能解决，你能帮帮她么？

Input

有多组测试数据;

第一行给你一个N(N<=400000)和一个M(M<=N)，N表示每个壁橱有N个方格,M表示要选出M种方案；

第二行和第三行分别有N个整数，表示每个方格的美味重量(结果保证在int以内)

Output

按照从小到大的顺序输出前M种方案的重量，每种方案占一行

Sample Input

2 21 23 4

Sample Output

45

 

通过构造得出N个序列的前K大，求所有序列中的前K大，使用优先队列维护一个长度为N的序列，时间复杂度为O(N*log(N))。

其实也是一个STL的问题。

AC代码：

#include<iostream> #include<set> #include<queue> #include<algorithm> #include<stdio.h> using namespace std; #define max 400008 int q[max],w[max];   struct node{     int c,d;     bool operator <(node p) const    {         return q[p.c]+w[p.d]<q[c]+w[d];     } }her;   priority_queue<node>myqueue; int main() {     int n,m;     while(~scanf("%d%d",&n,&m))     {         for(int i=0;i<n;i++) cin>>q[i];         for(int i=0;i<n;i++) cin>>w[i];         while(!myqueue.empty()) myqueue.pop();           sort(q,q+n);         sort(w,w+n);         for(int i=0;i<n;i++)         {             her.c=i;             her.d=0;             myqueue.push(her);         }         while(m--)         {             her=myqueue.top();             myqueue.pop();             printf("%d\n",q[her.c]+w[her.d]);             if(her.d != n-1){                 her.d++;                 myqueue.push(her);             }         }     }     return 0;           } 

路途中。。。。。