算法提高之LeetCode刷题LeetCode By GoLeetCode

4. 两个排序数组的中位数

2018-07-09  本文已影响4人  花果山松鼠

一、题目原型:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

二、题目意思剖析:

 nums1 = [1, 3]
 nums2 = [2]
 中位数是 2.0
 
 nums1 = [1, 2]
 nums2 = [3, 4]
 中位数是 (2 + 3)/2 = 2.5

三、解题思路:

1.先将数组排好序
2.如果数组count是奇数,那中间那位数(count / 2)就是中位数。
3.如果数组count是偶数,那中位数 = (第(count / 2) - 1 + 第(count / 2))/ 2
例子:
[20、21、23、23、25、29、32]
7 / 2 = 3,中位数 = 23
[20、21、23、23、25、29、32、33]
8 / 2 = 4,中位数 =(23+25)/ 2

func findMedianSortedArrays(_ nums1: [Int], _ nums2: [Int]) -> Double {
    var nums: [Int] = []
    var result: Double = 0
    for i in 0..<nums1.count {
        nums .append(nums1[i])
    }
    for i in 0..<nums2.count {
        nums .append(nums2[i])
    }
    
    nums.sort()
    print(nums)
    
    if nums.count % 2 == 0 { //偶数
       
        let num1: Int = nums[nums.count / 2 - 1]
        let num2: Int = nums[nums.count / 2]
        result = Double(Double(num1 + num2) / 2)
    }else if nums.count % 2 != 0 { // 奇数
        result = Double(nums[nums.count / 2])
    }
    return result;
}

四、小结

总通过数.png 提交结果.png

其实这是想到的最简答的方法,耗时会偏长,大家有好的想法可以留言啊,我暂时就没去深究了,继续下题下题。。。

有任何疑问都可以留言,非常乐意一起探讨。😄

上一篇 下一篇

猜你喜欢

热点阅读