LeetCode 第四题:寻找两个正序数组的中位数 【4/1000 】【python + go】

👤作者介绍:10年大厂数据\经营分析经验,现任大厂数据部门负责人。
会一些的技术:数据分析、算法、SQL、大数据相关、python
作者专栏每日更新:
LeetCode解锁1000题:打怪升级之旅
python数据分析可视化:企业实战案例
备注说明:方便大家阅读,统一使用python,带必要注释,公众号 数据分析螺丝钉 一起打怪升级

题库中的“寻找两个正序数组的中位数”问题是一个经典的算法挑战。这个问题不仅出现在在线编程平台上,也是许多技术面试中的常见题目,值得深入探讨一下

问题描述

给定两个大小分别为 m 和 n 的正序(非递减)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。示例:

输入:nums1 = [1,3], nums2 = [2]
输出:2.0
解释:合并数组 = [1,2,3],中位数 2

解题思路

理解这个问题的关键在于,两个数组是有序的,这为我们提供了很多可以利用的性质。一种直观的解法是将两个数组合并成一个大的有序数组,然后直接找到中位数。虽然这种方法很直接,但它的时间复杂度为 (O(m+n)),并不是最优解。

最优解法 —— 二分查找法

要提高效率,可以考虑二分查找法。核心思路是,中位数将一个集合分为两个长度相等的子集,一个子集中的所有元素都小于另一个子集中的元素。对于两个有序数组,我们可以利用二分查找在 (O(\log(m+n))) 的时间复杂度内找到中位数。

算法步骤

确定较短的数组:为了减少查找的范围,我们总是在两个数组中较短的那个进行二分查找。
二分查找:在较短数组中进行二分查找,确定一个切点,使得两个数组被分为左右两个部分,左边部分的元素总数等于右边部分(或者多一个,如果总元素数为奇数)。
调整切点:通过比较两个数组在切点附近的元素,调整切点的位置,直到满足中位数的性质:左边部分的最大元素小于或等于右边部分的最小元素。
计算中位数:根据切点确定的左右两部分,计算中位数。

python示例

def find_median_sorted_arrays(nums1: [int], nums2: [int]) -> float:# 确保 nums1 是两个中较短的数组if len(nums1) > len(nums2):nums1, nums2 = nums2, nums1m, n = len(nums1), len(nums2)# 初始化二分查找的边界left, right, half_len = 0, m, (m + n + 1) // 2while left <= right:# i 和 j 分别为两个数组的切分点i = (left + right) // 2j = half_len - i# 调整左边界if i < m and nums2[j-1] > nums1[i]:left = i + 1# 调整右边界elif i > 0 and nums1[i-1] > nums2[j]:right = i - 1else:# 找到合适的切分点,计算中位数if i == 0: max_of_left = nums2[j-1]elif j == 0: max_of_left = nums1[i-1]else: max_of_left = max(nums1[i-1], nums2[j-1])if (m + n) % 2 == 1:# 如果 m + n 是奇数,中位数是左半部的最大值return max_of_left# 如果 m + n 是偶数,中位数是左半部的最大值和右半部的最小值的平均值if i == m: min_of_right = nums2[j]elif j == n: min_of_right = nums1[i]else: min_of_right = min(nums1[i], nums2[j])return (max_of_left + min_of_right) / 2.0return 0# 示例测试
nums1 = [1, 3]
nums2 = [2]
print(f"中位数是:{find_median_sorted_arrays(nums1, nums2)}")

代码解读:

让我们一步一步分析上面的Python代码,以解释它是如何找到两个有序数组中位数的。

关键变量

  • nums1 和 nums2: 输入的两个有序数组。
  • m 和 n: 分别是 nums1 和 nums2 的长度。
  • left 和 right:用于在较短数组 nums1 上执行二分查找的指针。
  • half_len: 两个数组合并后左半部分的长度。

二分查找

二分查找的目的是在较短的数组 nums1 中找到一个位置 i,同时在 nums2 中找到位置 j,使得:

  • nums1[i-1] <= nums2[j] 且 nums2[j-1] <= nums1[I]

位置 i 和 j 的选取使得nums1和nums2被分割成两个部分,左半部分的所有元素都不大于右半部分的元素。

边界条件处理

循环中的条件判断用于处理边界条件,确保不会访问数组外的元素:

  • 如果 i 还可以增大,并且 nums2[j-1] > nums1[i],那么需要将 left 移动到 i + 1,这意味着 i 的位置太小,需要向右移动以增大 i。
  • 如果 i 还可以减小,并且 nums1[i-1] > nums2[j],那么需要将 right 移动到 i - 1,这意味着 i 的位置太大,需要向左移动以减小 i。

计算中位数

一旦找到了合适的 i 和 j,中位数可以通过下列方式计算:

  • 当 m + n 为奇数时,中位数是左半部分的最大值。
  • 当 m + n 为偶数时,中位数是左半部分的最大值和右半部分的最小值的平均值。
    代码中使用了几个条件判断来处理这个逻辑,确保即使在边界条件下(比如某一数组的所有元素都在另一数组的左边或右边时)也能正确计算中位数。

返回结果

最后,根据 m + n 的奇偶性,返回合适的中位数值。

这个算法精妙地利用了二分查找和有序数组的性质来高效地解决问题,避免了将两个数组合并后再查找中位数,从而达到了更优的时间复杂度 O(log(min(m,n)))。

goland示例

我们同样可以采用二分查找法。以下是 Go 语言版本的实现及其解读。

package mainimport ("fmt""math"
)func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {// 确保 nums1 是较短的数组if len(nums1) > len(nums2) {nums1, nums2 = nums2, nums1}m, n := len(nums1), len(nums2)left, right, halfLen := 0, m, (m+n+1)/2var maxOfLeft, minOfRight intfor left <= right {i := (left + right) / 2j := halfLen - iif i < m && nums2[j-1] > nums1[i] {left = i + 1} else if i > 0 && nums1[i-1] > nums2[j] {right = i - 1} else {if i == 0 {maxOfLeft = nums2[j-1]} else if j == 0 {maxOfLeft = nums1[i-1]} else {maxOfLeft = int(math.Max(float64(nums1[i-1]), float64(nums2[j-1])))}if (m+n)%2 == 1 {return float64(maxOfLeft)}if i == m {minOfRight = nums2[j]} else if j == n {minOfRight = nums1[i]} else {minOfRight = int(math.Min(float64(nums1[i]), float64(nums2[j])))}return float64(maxOfLeft+minOfRight) / 2.0}}return 0.0
}func main() {nums1 := []int{1, 3}nums2 := []int{2}fmt.Println("中位数是:", findMedianSortedArrays(nums1, nums2))
}

代码解读

  • 此解决方案首先检查 nums1 和 nums2 的长度,确保 nums1 是较短的数组,这样可以减少我们二分查找的范围。
  • left 和 right 变量定义了 nums1 上的查找范围,而 halfLen 变量则根据两数组的总长度计算中位数左侧的元素数量。
  • 在二分查找过程中,i 和 j 分别表示 nums1 和 nums2 中用于分割数组的索引。left 和 right 的调整基于 nums1[i] 和 nums2[j] 的比较,以确保左侧的最大值不大于右侧的最小值。
  • 当找到合适的分割线时,根据奇偶性计算中位数。如果总长度是奇数,则中位数是左侧的最大值;如果是偶数,则中位数是左侧最大值和右侧最小值的平均数。

总结

通过这种方法,我们能够在
O(log(min(m,n))) 时间复杂度内找到两个有序数组的中位数。Go 语言的实现利用了数学库中的 math.Max 和 math.Min 函数来简化代码,同时保持了算法的核心逻辑清晰明了。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/789535.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

BioTech - AlphaFlow 项目 PyTorch Lightning + DeepSpeed 的分布式配置

欢迎关注我的CSDN:https://spike.blog.csdn.net/ 本文地址:https://blog.csdn.net/caroline_wendy/article/details/137348092 GitHub: https://github.com/bjing2016/alphaflow 步骤如下: 安装 DeepSpeed:需要安装 DeepSpeed。配置 Trainer:在 PyTorch Lightning 中,需要…

有关数据开发项目中使用HIVE由于无法update和delete的场景下,如何解决数据增量的思路

解决数据增量问题的思路在Hive中 在数据开发项目中&#xff0c;使用Hive进行数据处理时&#xff0c;由于Hive不支持update和delete语句&#xff0c;处理数据增量可能会变得有些棘手。然而&#xff0c;有几种策略和技术可以帮助我们解决这个问题&#xff0c;并确保数据增量的高…

ELK报错,索引变成只读状态。

问题描述 今天发现当天的索引在ES中并没有创建&#xff0c;logstash中不停的报错&#xff1a; [2021-05-24T05:47:51,904][INFO ][logstash.outputs.elasticsearch] retrying failed action with response code: 403 ({“type”>“cluster_block_exception”, “reason”&g…

3D检测:从pointnet,voxelnet,pointpillar到centerpoint

记录centerpoint学习笔记。目前被引用1275次&#xff0c;非常高。 地址&#xff1a;Center-Based 3D Object Detection and Tracking (thecvf.com) GitHub - tianweiy/CenterPoint CenterPoint&#xff1a;三维点云目标检测算法梳理及最新进展&#xff08;CVPR2021&#xff…

rust- 结构体与二进制数组转换

将结构体当二进制流传输是做网络编程时传输协议的常用功能。golang语言可以使用包 encoding/binary实现&#xff0c;例如 import ("encoding/binary""os" )... err : binary.Write(f, binary.LittleEndian, p) ...rust中可以使用 deku将结构体实例转换为by…

openGauss 级联备机

级联备机 可获得性 本特性自openGauss 1.1.0版本开始引入。 特性简介 本特性主要基于当前一主多备的架构&#xff0c;在此基础上&#xff0c;支持级联备机连接备机。 客户价值 一主多备架构在特殊业务场景下&#xff0c;无法支持足够灵活的结构。多机房部署&#xff0c;不…

STM32 TIM DMA burst 输出变频 PWM 波形

1. 问题背景 客户需要 MCU 输出一组变频的 PWM 波形来控制外围器件&#xff0c;并且不同频率脉冲的个数也不同。STM32U5 芯片拥有 TIM1/TIM8 高级定时器&#xff0c;还有通用定时器TIM2/TIM3/TIM4/TIM5 以及 TIM15/TIM16/TIM17。TIM 模块中&#xff0c;可通过修改 ARR 寄存器的…

一文让你彻底理解 AdaBoost 自适应提升算法 | AdaBoost 的关键要点、基本原理、优缺点和实际应用

&#x1f349; CSDN 叶庭云&#xff1a;https://yetingyun.blog.csdn.net/ 一、关键要点 AdaBoost&#xff0c;即自适应提升&#xff08;Adaptive Boosting&#xff09;算法的缩写&#xff0c;是一种基于 Boosting 策略的集成学习方法&#xff0c;旨在降低偏差。AdaBoost 的 “…

Vue和FastAPI实现前后端分离

前言 近期接触了一些开源大模型应用服务&#xff0c;发现很多用的都是FastAPI web框架&#xff0c;于是乎研究了一下它的优势&#xff0c;印象最深有两个&#xff1a;一个是它的异步处理性能比较好&#xff0c;二是它可以类似java swagger的API交互文档&#xff0c;这个对应前…

【快捷部署】013_Podman(3.4.4)

&#x1f4e3;【快捷部署系列】013期信息 编号选型版本操作系统部署形式部署模式复检时间013podman3.4.4Ubuntu 22.04apt-2024-04-03 一、快捷部署 注意! 必须满足&#xff1a;Ubuntu 20.10 and newer #由于本期安装脚本较为简单&#xff0c;所以不制作一键安装脚本&#xf…

【Django学习笔记(三)】BootStrap介绍

BootStrap介绍 前言正文1、BootStrap 快速了解2、初识BootStrap2.1 下载地址2.2 创建目录2.3 引入BootStrap2.4 使用BootStrap 3、BootStrap 组件&样式3.1 导航条3.2 栅格系统3.3 container3.3.1 container3.3.2 container-fluid 3.4 面板3.5 媒体对象3.6 分页3.7 图标3.7.…

SAP-CO主数据之作业类型创建-<KL01>

公告&#xff1a;周一至周五每日一更&#xff0c;周六日存稿&#xff0c;请您点“关注”和“在看”&#xff0c;后续推送的时候不至于看不到每日更新内容&#xff0c;感谢。 目录 一、背景&#xff1a; 成本中心主数据创建&#xff1a; 成本要素主数据创建&#xff1a; 二…

氟化氢冷凝装置PFA反应烧瓶可应用半导体行业

PFA多颈烧瓶是一种高品质的实验室器皿&#xff0c;它具有多个颈口&#xff0c;可以在同一容器内进行多种化学反应。PFA多颈烧瓶能够耐受高温和强酸、强碱等腐蚀性介质&#xff0c;是化学实验中不可或缺的物品。多颈烧瓶可以配合搅拌桨、温度计、恒压分液漏斗、冷凝管等使用&…

golang语言系列:学习路线图

云原生学习路线导航页&#xff08;持续更新中&#xff09; 本文是 golang语言系列 文章&#xff0c;主要展示golang语言学习的全路线图 参考&#xff1a;https://github.com/darius-khll/golang-developer-roadmap/blob/master/i18n/zh-CN/ReadMe-zh-CN.md

宁波中墙建材水泥用半挂车运输,如何卸货,有什么注意点

宁波中墙建材水泥用半挂车运输&#xff0c;如何卸货&#xff0c;有什么注意点 水泥用半挂车运输卸货时&#xff0c;一般可以采用以下步骤&#xff1a; 准备工作&#xff1a;确保卸货区域安全&#xff0c;清理障碍物&#xff0c;检查卸货设备是否正常。停车就位&#xff1a;将半…

本地Windows打包启动前端后台

本地Windows打包启动前端后台 1、安装jdk Windows JDK安装 2、Nginx 2.1、将 nginx-1.16.1文件夹复制到D:\home\jisapp目录下 2.2、域名证书配置&#xff1a; 将域名证书放到D:\home\jisapp\ssl\2023目录下->配置nginx.conf文件&#xff08;D:\home\jisapp\nginx-1.22.0…

通过nvtx和Nsight Compute分析pytorch算子的耗时

通过nvtx和Nsight Compute分析pytorch算子的耗时 一.效果二.代码 本文演示了如何借助nvtx和Nsight Compute分析pytorch算子的耗时 一.效果 第一次执行,耗时很长 小规模的matmul,调度耗时远大于算子本身 大规模的matmul,对资源的利用率高小规模matmul,各层调用的耗时 二.代码…

【Linux】Vim编辑器

专栏文章索引&#xff1a;Linux 目录 在Vim编辑器中&#xff0c;一个Tab键相当于几个空格&#xff1f; 在Vim编辑器中&#xff0c;一个Tab键相当于几个空格&#xff1f; 在Vim编辑器中&#xff0c;默认情况下&#xff0c;一个Tab键相当于8个空格。 这是Vim的默认设置&#x…

Linux 设备驱动管理之内核对象(Kernel Object)机制

Linux 设备驱动管理之内核对象(Kernel Object)机制 Linux内核是一个复杂的系统&#xff0c;它通过一系列的机制和结构体来管理和表示系统中的资源。其中一个关键的概念是“内核对象”&#xff08;Kernel Object&#xff0c;简称kobject&#xff09;。本文将深入探讨kobject机制…

Springboot导出mysql数据到Excel表

Controller层代码 /*** 导出数据管理审核表*/ApiOperation(value "导出数据管理审核表")Log(title "导出数据管理审核表", businessType BusinessType.EXPORT)PostMapping("/exportCsv")public void exportCsv(HttpServletResponse response,…