P1631 序列合并 洛谷

https://www.luogu.org/problem/show?pid=1631

题目描述

有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N^2个和，求这N^2个和中最小的N个。

输入输出格式

输入格式：

第一行一个正整数N；

第二行N个整数Ai,满足Ai<=Ai+1且Ai<=10^9;

第三行N个整数Bi, 满足Bi<=Bi+1且Bi<=10^9.

【数据规模】

对于50%的数据中，满足1<=N<=1000；

对于100%的数据中，满足1<=N<=100000。

输出格式：

输出仅一行，包含N个整数，从小到大输出这N个最小的和，相邻数字之间用空格隔开。

输入输出样例

输入样例#1：

3
2 6 6
1 4 8

输出样例#1：

3 6 7

 #include <algorithm>
 #include <iostream>
 #include <cstdio>
 #define N 100015
 
 using namespace std;
 
 int n,size,a[N],b[N];
 int heap[N],ans[N];
 
 void push(int x)
 {
     heap[++size]=x;
     int now=size,next;
     while(now>)
     {
         next=now/;
         if(heap[next]>=heap[now])    break;
         swap(heap[next],heap[now]);
         now=next;
     }
 
 }
 
 void pop()
 {
     heap[]=heap[size--];
     int now=,next;
     while(now*<=size)
     {
         next=now*;
         if(next<size&&heap[next]<heap[next+])    next++;
         if(heap[now]>=heap[next])    break;
         swap(heap[now],heap[next]);
         now=next;
     }
 }
 
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++)    scanf("%d",&a[i]);
     for(int i=;i<=n;i++)    scanf("%d",&b[i]);
     sort(a+,a+n+);
     sort(b+,b+n+);
     for(int i=;i<=n;i++)    push(a[]+b[i]);
     for(int i=;i<=n;i++)
         for(int j=;j<=n;j++)
         {
             if(heap[]>a[i]+b[j])
             {
                 pop();
                 push(a[i]+b[j]);
             }
             else break;
         }
     for(int i=;i<=n;i++)
     {
         ans[i]=heap[];
         pop();
     }
     while(n--)
     {
         printf("%d ",ans[n+]);
     }
     return ;
 }

优先队列T2ge