CDOJ 1329 卿学姐与魔法（优先队列）

卿学姐与魔法

Time Limit: 1200/800MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

Submit  Status

“你的膜法也救不了你”——蛤

在去拯救公主的道路上，卿学姐披荆斩棘，刀刃早已锈迹斑斑。

一日卿学姐正在为武器的问题发愁，碰到了正在赏树的天行廖。

天行廖嘴角微扬，似乎看穿了卿学姐的心思，故意在此等待。

“少年，你渴望掌握雷电的力量吗？”天行廖如是问道。

已经差不多是条咸鱼的卿学姐欣然答应了。于是卿学姐开始跟随魔法大师天行廖学习魔法的力量。

刚入门的卿学姐发现，每个魔法都是由两种基本元素构成的，A元素和B元素。

而每个魔法的魔力是合成这个魔法的A元素和B元素的大小的和。

例如一个大小为3的A元素和一个大小为6的B元素，能构成一个魔力为9的魔法。

现在卿学姐收集了 N N个A元素和 N N个B元素。

敏锐的卿学姐立刻发现他能组合出 N∗N N∗N种魔法。

谦虚的卿学姐并不希望自己太跳，所以他准备将这 N∗N N∗N种魔法中的最小的 N N种展示给天行廖检查。

现在卿学姐想知道，这 N∗N N∗N种魔法中最小的 N N种是什么。

当然，得从小到大输出哦~

Input

第一行一个整数 N N

接下来一行有 N N个数，表示 N N个A元素

接下来一行有 N N个数，表示 N N个B元素

1≤N≤100000 1≤N≤100000

1≤A[i],B[i]≤1000000000 1≤A[i],B[i]≤1000000000

Output

输出 N N行，每行一个整数

代表 N∗N N∗N种魔法中最小的 N N个

Sample input and output

Sample Input	Sample Output
5 1 3 2 4 5 6 3 4 1 7	2 3 4 4 5

Source

2016 UESTC Training for Data Structures

题解：对两个序列分别排序，使用优先队列存储元素乘积，如果大于现在的队首元素，则跳出循环

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>

using namespace std;

priority_queue<int> p;

int main()
{
    int num[100000],A[100000],B[100000],N,q,j;
    scanf("%d",&N);
    for(int i = 0 ;i<N;i++)
        scanf("%d",&A[i]);
    sort(A,A+N);
    for(int i = 0 ;i<N;i++)
        scanf("%d",&B[i]);
    sort(B,B+N);
    for(int i = 0 ;i<N;i++)
        p.push(A[0] + B[i]);
    for(int i = 1 ;i<N;i++)
    {
        if(A[i]+B[0]>p.top())
            break;
        for(int j = 0;j<N;j++)
        {
           if(A[i]+B[j]<p.top())
            {
                p.pop();
                p.push(A[i]+B[j]);
            }
            else
                break;
        }
    }
    for(int i = 0;i<N;i++)
    {
        num[i] = p.top();
        p.pop();
    }
    for(int i = N-1;i>=0;i--)
    {
        printf("%d\n",num[i]);
    }
    return 0;
}

，