【ybt】【数据结构二分堆课过例2】序列合并

最新推荐文章于 2021-07-09 10:49:03 发布

原创最新推荐文章于 2021-07-09 10:49:03 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

ybt 同时被 2 个专栏收录

91 篇文章

订阅专栏

堆

4 篇文章

订阅专栏

本文介绍了如何使用贪心算法解决序列合并问题，通过建立二分堆来存储数列a和b的元素，每次取出最小值并推进数列a，重复此过程以合并两个数列。

序列合并

题目链接：序列合并

题目描述

在这里插入图片描述

解题思路

我们可以把它当成贪心来想：
首先将 $a$ 数列的第一个与 $b$ 数列的每一个匹配，用堆储存，每次取最小值，在将 $a$ 数列向后移一位进行匹配，重复 $n$ 遍。

code

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;

priority_queue<pair<int,pair<int,int> > > q;

int n;
int a[100010];
int b[100010];

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]);
	for(int i=1;i<=n;i++)
		q.push(make_pair(-a[i]-b[1],make_pair(i,1)));
	for(int i=1;i<=n;i++)
	{
		int x=q.top().second.first;
		int y=q.top().second.second;
		cout<<-q.top().first<<" ";
		q.pop();
		q.push(make_pair(-a[x]-b[y+1],make_pair(x,y+1)));
	}
}