YbtOJ 数据结构课堂过关例2 序列合并【大根堆】※

最新推荐文章于 2024-09-09 17:38:55 发布

JA_yichao

最新推荐文章于 2024-09-09 17:38:55 发布

阅读量138

点赞数

分类专栏：题解二叉堆 YbtOJ专项练习题文章标签：数据结构二叉堆算法题解 YbtOJ

本文链接：https://blog.csdn.net/Jackma_mayichao/article/details/116405987

版权

题解同时被 3 个专栏收录

414 篇文章 2 订阅

订阅专栏

YbtOJ专项练习题

102 篇文章 2 订阅

订阅专栏

二叉堆

4 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路

这道题我们可以从题目中得到一个简单的规律：
$a 1 + b 1 < a 1 + b 2 < a 1 + b 3$
$a 2 + b 1 < a 2 + b 2 < a 2 + b 3$
$a 3 + b 1 < a 3 + b 2 < a 3 + b 3$
我们可以从第一列中找一个最小的和，记录答案。
然后从答案的位置往后推一个和，再与前面的答案进行比较。
用样例来举例，顺序就是：
$\max(a1+b1,a2+b1,a3+b1)\to 3$
$\max(a1+b2,a2+b1,a3+b1)\to 6$
$\max(a1+b3,a2+b1,a3+b1)\to 7$
其实就是一个贪心。

代码

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
int n,a[100010],b[100010];
priority_queue<pair<int,pair<int,int> > > q;
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,y=q.top().second.second;
	 	q.pop();
	 	cout<<a[x]+b[y]<<" ";
	 	q.push(make_pair(-a[x]-b[y+1],make_pair(x,y+1)));
	 }
	return 0;
}