1. 程式人生 > >最小的N個和(優先隊列)

最小的N個和(優先隊列)

包含 containe contain content 優先 cin 得到 bool end

描述

有兩個長度為N的序列A和B,在A和B中各任取一個數相加可以得到N^2個和,求這N^2個和中最小的N個。

輸入

第一行輸入一個正整數N(N<=10^5);第二行N個整數Ai且Ai<=10^9;第三行N個整數Bi且Bi<=10^9。

輸出

輸出僅一行,包含n個整數,從小到大輸出這n個最小的和,相鄰數字之間用空格隔開。

輸入樣例 1

5
1 3 2 4 5
6 3 4 1 7

輸出樣例 1

2 3 4 4 5

輸入樣例 2

10
-1 3 2 0 12 6 2 3 -5 5
-3 2 5 1 0 12 32 -2 2 3

輸出樣例 2

-8 -7 -5 -4 -4 -3 -3 -3 -3 -2
 1 #include <bits/stdc++.h>
 2 
 3 using namespace std;
 4 long long a[100005];
 5 long long b[100005];
 6 
 7 struct note{
 8     int num,y;
 9     bool operator <(const note&a)const{
10             return num>a.num;
11     }
12 };
13
14 priority_queue <note> q; 15 16 int main() 17 { 18 int n; 19 cin>>n; 20 int d; 21 note k; 22 for(int i=1;i<=n;i++) 23 cin>>a[i]; 24 for(int i=1;i<=n;i++) 25 cin>>b[i]; 26 sort(a+1,a+n+1,less<int>()); 27 sort(b+1
,b+n+1,less<int>()); 28 for(int i=1;i<=n;i++) 29 { 30 k.y=1; 31 k.num=a[i]+b[1]; 32 q.push(k); 33 } 34 int flag=0; 35 for(int i=1;i<=n;i++) 36 { 37 k=q.top(); 38 q.pop(); 39 d=k.num; 40 int y=k.y; 41 k.num=d-b[y]+b[y+1]; 42 k.y=y+1; 43 q.push(k); 44 if(flag==0) 45 { 46 cout<<d; 47 flag=1; 48 } 49 else 50 cout<<" "<<d; 51 } 52 cout<<endl; 53 return 0; 54 }

最小的N個和(優先隊列)