2016 UESTC Training for Data Structures D - 卿学姐与魔法 优先队列、构造

D - 卿学姐与魔法

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

Submit  Status

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Input

第一行一个整数 N

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

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

1≤N≤100000                                    

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

Output

输出 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

Hint

注意可能会爆int哦

Source

2016 UESTC Training for Data Structures  Problem D

My Solution

用STL里的优先队列直接维护前n+1小的数字

ptr = 0

先 i = ptr 然后A + B1 + B2 + …… B(n-1)

然后 j = 1 B1 + A2 + ……A(n-1)

这样依次下去，

当 Aptr + B(ptr+1) 已经比.top()大了，则可以break了。

故最好的情况O(n), 最差的情况(┬＿┬)不会算但绝对小于O(n^2) 

竟然用了do while 开心

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int maxn = 100000+8;

int A[maxn], B[maxn];
priority_queue<int> pq;
vector<int> vec;
//2*sort This has been test in A + B problem
int main()
{
    #ifdef LOCAL
    freopen("a.txt", "r", stdin);
    #endif // LOCAL
    int n;
    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);

    int cnt = 0;     // 0 or 1 just see the test ☺
    int ptr = 0;
    do{
        for(int j = ptr; j < n ; j++){
            if(cnt != n+1){cnt++; pq.push(A[ptr]+B[j]);}
            else{
                if(pq.top() > A[ptr]+B[j]){
                    pq.push(A[ptr]+B[j]);
                    pq.pop();
                }
                else break;
            }
        }

        for(int j = ptr+1; j < n ; j++){
            if(cnt != n+1){cnt++; pq.push(A[j]+B[ptr]);}
            else{
                if(pq.top() > A[j]+B[ptr]){
                    pq.push(A[j]+B[ptr]);
                    pq.pop();
                }
                else break;
            }
        }
        ptr++;
    } while(pq.top() >= A[ptr]+B[ptr] && ptr < n);
    pq.pop();
    while(!pq.empty()){
        vec.push_back(pq.top());
        pq.pop();
    }
    for(int i = n-1; i >= 0; i--)
        printf("%d\n", vec[i]);
    return 0;
}

Thank you!

                                                                                                                                               ------from  ProLights