首页 > 代码库 > wikioi 1245 小顶堆
wikioi 1245 小顶堆
题目描述 Description
有两个长度为 N 的序列 A 和 B,在 A 和 B 中各任取一个数可以得到 N^2 个和,求这N^2 个和中最小的 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。
这题刘汝佳白书中190页里有,挺好的堆思想题。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<map> #include<queue> #include<set> #include<bitset> #define INF 100007 using namespace std; typedef unsigned long long ull; struct abc { int sum,i; abc(int sum,int i):sum(sum),i(i){} bool operator<(abc a)const { return sum>a.sum; } }; priority_queue<abc>heap; int a[100010],b[100010]; int main() { int n,i,j; cin>>n; for(i=0;i<n;i++) scanf("%d",a+i); for(i=0;i<n;i++) scanf("%d",b+i); sort(a,a+n); sort(b,b+n); for(i=0;i<n;i++) heap.push(abc(a[i]+b[0],0)); for(i=0;i<n;i++) { abc it=heap.top();heap.pop(); printf("%d%c",it.sum,i==n-1?'\n':' '); if(it.i+1<n) heap.push(abc(it.sum-b[it.i]+b[it.i+1],it.i+1)); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。