LeetCode刷题——探寻两个有序数组中位数

探寻两个有序数组中位数题目的思路探讨与源码
探寻两个有序数组中位数的题目如下图,该题属于数组类的题目,主要考察对于中位数的理解,通过考虑不同两个数组的相对位置情况,最终求出中位数。本人的思路比较懒,直接使用了列表的合并,该题虽然属于hard难度但是解决思路较简单。
在这里插入图片描述
本人在看到该题目后,认为需要将两个数组也就是列表进行合并,不能使用set集合因为会去除两个数组的重复元素,所以使用列表进行合并,然后进行判断,最终的列表究竟是奇数个元素还是偶数个元素,分别求中位数返回即可。代码如下所示:

#喷火龙与水箭龟
nums1=[1,3]
nums2=[2]
N1=len(nums1)
N2=len(nums2)
mid=0
if(N1==0 or N2==0):
	if(N1==0):
		midIndex=int(N2/2)
		if(N2%2==0):
			mid=float((nums2[midIndex]+nums2[midIndex-1])/2)
			print(mid)
		else:
			mid=float(nums2[midIndex])
			print(mid)
	if(N2==0):
		midIndex=int(N1/2)
		if(N1%2==0):
			mid=float((nums1[midIndex]+nums1[midIndex-1])/2)
			print(mid)
		else:
			mid=float(nums1[midIndex])
			print(mid)
setList=nums1+nums2
setList.sort()
Nf=len(setList)
midIndex=int(Nf/2)
if(Nf%2==0):
	mid=float((setList[midIndex]+setList[midIndex-1])/2)
	print(mid)
else:
	mid=float(setList[midIndex])
	print(mid)

在上述代码中,setList是合并后的数组,Nf是合并后的数组长度,midIndex是中位数的下标,通过判断偶数个元素或者奇数个元素进行计算,最终得到结果。最终输入是nums1=[1,3] nums2=[2] ,得到结果2.0 ,是正确的。当然个人认为可能最前面的代码可以省略,那段代码是用来判别是否有一个元素可能为空的情况。需要注意的是,本代码是作者的测试代码,如需放到LeetCode上运行,需要进行函数缩进和加上返回值才可以运行。
最终结果如下:
在这里插入图片描述
从结果来说是速度还行,但是应该可以进一步提速,希望朋友们能够多多指教,非常感谢。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值