Swift 力扣第四题：寻找两个有序数组的中位数

题目：

4. 寻找两个有序数组的中位数
 难度

 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

 请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

 你可以假设 nums1 和 nums2 不会同时为空。

 示例 1:

 nums1 = [1, 3]
 nums2 = [2]

 则中位数是 2.0
 示例 2:

 nums1 = [1, 2]
 nums2 = [3, 4]

 则中位数是 (2 + 3)/2 = 2.5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/longest-substring-without-repeating-characters
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答一：暴力。。

class Solution {
    func findMedianSortedArrays(_ nums1: [Int], _ nums2: [Int]) -> Double {
        let arr = (nums1 + nums2).sorted()// sorted 比冒泡排序简单高效
        let count = arr.count
        let index = count / 2
        if count % 2 > 0 {
            return Double(arr[index])
        } else {
            let num1 = Double(arr[index])
            let num2 = Double(arr[index - 1])
            return (num1 + num2) / 2
        }
    }
}

执行结果：通过
执行用时 : 112 ms, 在所有 Swift 提交中击败了58.52%的用户
内存消耗 : 21.3 MB, 在所有 Swift 提交中击败了26.92%的用户

很不理想
解答二：

class Solution {
    func findMedianSortedArrays(_ nums1: [Int], _ nums2: [Int]) -> Double {
        let len1 = nums1.count
        let len2 = nums2.count
        
        var nums1 = nums1
        var nums2 = nums2
        if len1 == 0 {
            nums1 = nums2
        }
        if len2 == 0 {
            nums2 = nums1
        }
        
        if len1 == 1 && len2 == 1 {
            return Double(nums1[0] + nums2[0]) / 2
        }
        
        let lenAll = len1 + len2
        let mid = lenAll / 2 + 1
        var i = 0, j = 0
        var array = [0, 0]
        
        while i < len1 && j < len2 {
            if nums1[i] < nums2[j] {
                array[(i + j) % 2] = nums1[I]
                i += 1
            } else {
                array[(i + j) % 2] = nums2[j]
                j += 1
            }
            if i + j == mid {
                break
            }
        }

        while i + j < mid && i == len1 {
            array[(i + j) % 2] = nums2[j]
            j += 1
        }
        while i + j < mid && j == len2 {
            array[(i + j) % 2] = nums1[I]
            i += 1
        }
                
        if lenAll % 2 != 0 {
            return Double(array[(i + j - 1) % 2])
        } else {
            return Double(array[0] + array[1]) / 2
        }
    }
}