本文主要是介绍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;
}
这篇关于wikioi 1245 小顶堆的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!