最小的N個和(優先隊列)
阿新 • • 發佈:2019-03-27
包含 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 }; 1314 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個和(優先隊列)