P1631 序列合并

Posted 2021-01-10 tony-double-sky

P1631 序列合并

题目描述
有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N^2个和，求这N^2个和中最小的N个。
对于100%的数据中，满足1<=N<=100000。

Solution

数据范围表示 (N^{2}) 要命
然而有一个想法， 要找出这 (N^{2}) 个数的大致大小情况， 从这里入手

对两序列排序
我们把这 (N^{2}) 个数列出来， 可以发现， 以下式子成递增序列
一行中前一个恒大于等于后一个
[a_{1} + b_{1}, a_{1} + b_{2}, ..., a_{1} + b_{N}]
[a_{2} + b_{1}, a_{2} + b_{2}, ..., a_{2} + b_{N}]
[...]
[a_{N} + b_{1}, a_{N} + b_{2}, ..., a_{N} + b_{N}]

所以我们先把每一行第一个放入小根堆中， 每次取出堆顶输出， 同时将此行下一个数丢进堆中， 重复 (N) 次即可， 复杂度 (O(Nlog N))

Code

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<climits>
#include<queue>
#define LL long long
using namespace std;
int RD(){
    int out = 0,flag = 1;char c = getchar();
    while(c < '0' || c >'9'){if(c == '-')flag = -1;c = getchar();}
    while(c >= '0' && c <= '9'){out = out * 10 + c - '0';c = getchar();}
    return flag * out;
    }
const int maxn = 200019;
int num;
int a[maxn], b[maxn];
struct Node{
    int val, i, j;
    bool operator < (const Node &a)const{
        return val < a.val;
        }
    };
priority_queue<Node>Q;
int main(){
    num = RD();
    for(int i = 1;i <= num;i++)a[i] = RD();
    for(int i = 1;i <= num;i++)b[i] = RD();
    sort(a + 1, a + 1 + num), sort(b + 1, b + 1 + num);
    for(int i = 1;i <= num;i++)Q.push((Node){-(a[i] + b[1]), i, 1});
    while(num--){
        Node now = Q.top();Q.pop();
        int i = now.i, j = now.j;
        printf("%d ", -now.val);
        Q.push((Node){-(a[i] + b[j + 1]), i, j + 1});
        }
    puts("");
    return 0;
    }