合并K个有序数组

这篇博客探讨了如何合并K个有序数组的问题,提供了三种解法:直接排序、归并排序和构造小根堆。重点分析了构造小根堆的解题思路,包括遍历所有数组确定元素总数、创建小根堆、不断更新堆并取出最小元素的过程。这种方法的时间复杂度为O(mnlogn),其中m为元素总个数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在之前的面试中有遇到这个问题,当时答得不是很满意,所以,下来之后把这个题完整的写了一下。
问题:有K个长度为N的有序数组,将着这些个数组合并为一个有序的数组。
将原问题解决后,我又将原问题修改了一下:
有K个长度不定的有序数组,将这些数组合并为一个有序数组;

三种解法:

  1. 申请一个数组,将所有的元素添加到该数组中,使用sort()函数,直接进行排序;
  2. 归并排序;两两归并,使用merge()函数;
  3. 构造小根堆;

第一种解法的时间复杂度为:O(nlog2n)
第二种解法的时间复杂度为:O(knlogn)
第三种解法的时间复杂度为:O(mnlogn)(m为元素总个数)
第一二种解法,比较简单,这里不再赘述。下面只分析第三种解法;

解法三解题思路:
  1. 首先遍历所有的数组,确定最终所有元素的个数(count);
  2. 创建一个大小为count的数组用来保存最终结果,再创建一个大小为K的小根堆;(小根堆保存的元素类型,是结构体)
  3. 先将每个数组的第一个元素插入到小根堆中,进行一次堆排序;
  4. 重复下面的操作,直至元素全部插入结果集中:
    a. 取出堆顶元素(堆中最小元素),保存到结果中;
    b. 将堆顶元素所在数组的下一个元素替换到堆顶,如果堆顶元素是其所在数组的最后 一个,将堆顶元素替换成INT_MAX;
    c. 进行一次堆排序;

下面是该题的详细解法:

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

struct Node
{
   
	int val;
	int col;
	int row;
	Node(int _val, int _row, int _col) 
		:val(_val)
		,row(_row)
		,col(_col)
	{
   }
};

//向下调整
void Adjust(vector<Node*>& heap, int parent)
{
   
	size_t child = parent * 2
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值