Acwing: 146.序列 二叉堆算法

目录

AcWing 146. 序列

题目描述

算法介绍

题目思路

代码及注释

题目描述

给定 m 个序列，每个包含 n 个非负整数。

现在我们可以从每个序列中选择一个数字以形成具有 m 个整数的序列。

很明显，我们一共可以得到  个这种序列，然后我们可以计算每个序列中的数字之和，并得到 个值。

现在请你求出这些序列和之中最小的 n 个值。

输入格式

第一行输入一个整数 T，代表输入中包含测试用例的数量。

接下来输入 T组测试用例。

对于每组测试用例，第一行输入两个整数 m 和 n。

接下在 m 行输入 m 个整数序列，数列中的整数均不超过 10000。

输出格式

对于每组测试用例，均以递增顺序输出最小的 n 个序列和，数值之间用空格隔开。

每组输出占一行。

数据范围

0 < m ≤ 1000, 0 < n ≤ 2000；

输入样例：

1
2 3
1 2 3
2 2 3

输出样例：

3 3 4

算法介绍

该题运用的算法是二叉堆，

堆（Heap）是一类数据结构，它们拥有树状结构，且能够保证父节点比子节点大（或小）。当根节点保存堆中最大值时，称为大根堆；反之，则称为小根堆。

二叉堆（Binary Heap）是最简单、常用的堆，是一棵符合堆的性质的完全二叉树。它可以实现 O(log⁡n) 地插入或删除某个值，并且 O(1) 地查询最大（或最小）值。

题目思路

该题中最重要的思路是分组

该题要求从每个序列中选择一个数字以形成具有 m 个整数的序列，这题如果直接进行两两结合进行求，那么时间复杂度就是,那肯定不行，那么就先求两两序列中的组合在一起的数，得到个数，然后从中挑选出最小的个数组成最优的那一组；如果直接计算的话，那么时间复杂度就是m*n*n 那么也是会超时的，那么就想选一个算法来优化这个n*n的过程。这时我们就可以想到组合之后插入二叉堆中进行求每个序列中最小的那个，我们可以先对a数组进行排序，那么组合之后的队列的每一个队头的数都是最小的那个，那么每次找所i有队头中最小的那个，记录到c数组中，然后将这个队头弹出，一共找到n个数，然后结束操作，并将记录在c数组中的答案更新到a数组中，来进行下一步查找。

代码及注释

/*
	*时间:  2023年7月4日
	*题目： AcWing: 146.序列 
	*算法： 二叉堆 
*/

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;

typedef pair<int,int> PII;

const int N = 20010;
int n,m;
int a[N],b[N],c[N];

void merge()
{
	priority_queue<PII,vector<PII>,greater<PII>> p;
	
	for(int i =1 ; i <= n; i++) p.push({a[1] + b[i],1});  // 这个1就是记录该数字是加的a[1]; 
	
	int cnt = 1;
	while(cnt <= n)
	{
		auto t = p.top();
		p.pop();
		int s = t.first;
		int q = t.second;
		c[cnt++] = s;                // c是为了记录答案的 
		p.push({s-a[q]+a[q+1],q+1});   //s-a[q]+a[q+1],s是加了a[q]后的值，如果想加后面的值需要先将a[q]删除掉 
	}
	for(int i =1 ; i <= n; i++) a[i] = c[i];// 更新数据，每次记录的都是上一次进行过后的最优解 
}

int main (void)
{
	int T;
	cin >> T;
	while(T--)
	{
		scanf("%d %d",&m,&n);
		for(int i = 1; i <= n; i++)
		{
			scanf("%d",&a[i]);
		}
		sort(a+1,a+1+n);
		m--;
		while(m--)
		{
			for(int i = 1; i <= n; i++)
			{
				scanf("%d",&b[i]);
			}
			merge();
		}
		for(int i = 1; i <= n; i++) printf("%d ",a[i]);
		puts(""); 
	}
	
	return 0;
}

制作不易，留下点赞再走吧