探寻两个有序数组中位数题目的思路探讨与源码
探寻两个有序数组中位数的题目如下图,该题属于数组类的题目,主要考察对于中位数的理解,通过考虑不同两个数组的相对位置情况,最终求出中位数。本人的思路比较懒,直接使用了列表的合并,该题虽然属于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上运行,需要进行函数缩进和加上返回值才可以运行。
最终结果如下:
从结果来说是速度还行,但是应该可以进一步提速,希望朋友们能够多多指教,非常感谢。