一道优先队列的题...- 2013ACM多校联合(4)_NUN - ZZ的橱柜

Problem B: ZZ的橱柜

Time Limit: 1 Sec   Memory Limit: 64 MB
Submit: 345   Solved: 48 

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

http://www.acmore.net/problem.php?cid=1016&pid=1

     题意抽象...有两个数组A,B..求第M小A[ i ] + B[ j ]的大小....

     总觉得这题在哪里看到过...将A,B先排序..用一个结构体数组..记录每个A的元素已经加到B的那个数了.. 初始化将A中每个元素指向B的1...丢到优先队列里去..优先队列是优先A [ i ] + B [ j ]  的小值.....每次取出队列顶的结构体...其A[ i ] + B[ j ] 就是当前的最小值..将 j++后丢回优先队列..做M次就可以得到第M小的值了

Program:

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#define ll long long 
#define oo 1000000001
#define MAXN 400005
using namespace std;
int n,m,a[MAXN],b[MAXN];
struct node
{
       int ai,bi;
       bool operator < (node p) const
       {
              return a[p.ai]+b[p.bi]<a[ai]+b[bi];
       }
}h;
priority_queue<node> myqueue;
int main()
{     
    //   freopen("input.txt","r",stdin);   freopen("output.txt","w",stdout);   
       int i;
       while (~scanf("%d%d",&n,&m))
       {
              for (i=1;i<=n;i++) scanf("%d",&a[i]);
              for (i=1;i<=n;i++) scanf("%d",&b[i]);
              while (!myqueue.empty()) myqueue.pop();
              sort(a+1,a+1+n);
              sort(b+1,b+1+n);
              for (i=1;i<=n;i++) 
              {
                    h.ai=i;  h.bi=1;
                    myqueue.push(h);
              }
              while (m--)
              {
                    h=myqueue.top();
                    myqueue.pop();
                    printf("%d\n",a[h.ai]+b[h.bi]);
                    if (h.bi!=n)
                    {
                           h.bi++;
                           myqueue.push(h);
                    }
              }
       }
       return 0;
}