codevs1245 最小的N个和
输入描述 Input Description
第一行输入一个正整数N;第二行N个整数Ai 且Ai≤10^9;第三行N个整数Bi,
且Bi≤10^9
输出描述 Output Description
输出仅一行,包含 n 个整数,从小到大输出这 N个最小的和,相邻数字之间用
空格隔开。
样例输入 Sample Input
5
1 3 2 4 5
6 3 4 1 7
样例输出 Sample Output
2 3 4 4 5
数据范围及提示 Data Size & Hint
【数据规模】 对于 100%的数据,满足 1≤N≤100000。
思路
先把两个数组从小到大排个序
先求出a数组中每个数与b数组中第一个数的和,作为堆的初始值,建一个小根堆,以num为优先级,y其次
由于我们不确定是a中每个数加b中最小数比较小,还是b中每个数加a中最小数比较小
所以要让堆动起来,插入的元素就是a中当时的元素与b中下一个元素的和,每次弹出堆顶元素,重复n次即可
#include<iostream> #include<algorithm> #include<queue> using namespace std; int n,a[100001],b[100001]; 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; }