码迷,mamicode.com
首页 > 其他好文 > 详细

[洛谷P1631]序列合并

时间:2017-08-19 12:44:28      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:++   思路   class   typedef   pac   can   size   循环   序列   

题目大意:给你两个非降序序列a和b,每个序列n个数,在a和b各取一个数相加,能得到$n^2$个和,求这些和中最小的n个。

解题思路:我们先把a[1]+b[1],a[1]+b[2],a[1]+b[3]...a[1]+b[n]扔进一个小根堆里,每次取出一个数,设为a[i]+b[j],那么我们输出这个数,并把a[i+1]+b[j]放进堆里,循环n次就能得出答案。时间复杂度$O(n\log n)$。

我用pbds的优先队列+pair实现,用第二个数保存a[i]+b[j]的i值。

然而我第一次push时i值没有加1,于是WA。加了1就是AC╭(╯^╰)╮

C++ Code:

#include<ext/pb_ds/priority_queue.hpp>
#include<cstdio>
#include<functional>
using namespace std;
using namespace __gnu_pbds;
typedef pair<int,int> pr;
__gnu_pbds::priority_queue<pr,greater<pr> >q;
int n,a[100005],b[100005];
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i)scanf("%d",&a[i]);
	for(int i=1;i<=n;++i){
		scanf("%d",&b[i]);
		q.push(make_pair(a[1]+b[i],1));
	}
	bool space=false;
	for(int i=1;i<=n;++i){
		pr p=q.top();q.pop();
		if(space)putchar(‘ ‘);else space=true;
		printf("%d",p.first);
		q.push(make_pair(p.first-a[p.second]+a[p.second+1],p.second+1));
	}
	puts("");
	return 0;
}

 

[洛谷P1631]序列合并

标签:++   思路   class   typedef   pac   can   size   循环   序列   

原文地址:http://www.cnblogs.com/Mrsrz/p/7395794.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!