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

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

題目描述

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

第一行輸入一個正整數N(1<=N<=100000);

第二行N個整數Ai且Ai<=109;第三行N個整數Bi且Bi<=109。
輸出

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

5
1 3 2 4 5
6 3 4 1 7

樣例輸出

2 3 4 4 5

解析:用優先佇列,每一個A數組裡面的值都有一個值,當這個值被用了之後就轉換成這個值和B數組裡面的下一個值求和就好了

#include<bits/stdc++.h>
using namespace std; const int N=1e5+10 int n,a[N],b[N]; struct node { int y,num; bool operator <(const node &v)const { return num>v.num; } }; node k; priority_queue<node>q; int main() { cin>>n; for(int i=1;i<=n;i++) cin>>a[i]; for
(int i=1;i<=n;i++) cin>>b[i]; sort(a+1,a+n+1); sort(b+1,b+n+1); for(int i=1;i<=n;i++) { k.y=1;k.num=a[i]+b[1]; q.push(k); } int s=1; while(s<=n) { node now=q.top(); q.pop(); if(now.y+1<=n) { k.y=now.y+1
; k.num=now.num-b[now.y]+b[now.y+1]; q.push(k); } s++; cout<<now.num<<' '; } return 0; }