LeetCode | Median of Two Sorted Arrays

題目連接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

給定兩個大小爲 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

思路：

自己的思路（時間、空間複雜度高一些）：將兩個有序數組合併爲一個有序數組，如果合併後的數組長度爲奇數：則中位數爲數組中間的元素；如果合併後的數組爲偶數：中位數爲數組中間兩個元素的商

網上思路：是從兩個數組中依次移除最小最大的一對數，重複操作直到兩個數組共剩下1個或者2個數，即爲中位數。
 

 

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double result = 0.0;

        int[] arr = new int[nums1.length+nums2.length];
        int i=0,j=0,d=0;
        while (i<nums1.length && j<nums2.length){
            if (nums1[i] <= nums2[j]){
                arr[d] = nums1[i];
                d++;
                i++;
            }else {
                arr[d] = nums2[j];
                d++;
                j++;
            }
        }
        if (i==nums1.length){
            for (;d<arr.length;d++){
                arr[d] = nums2[j];
                j++;
            }
        }else {
            for (;d<arr.length;d++){
                arr[d] = nums1[i];
                i++;
            }
        }
        if (arr.length%2 ==0){
            result = (arr[arr.length/2] + arr[arr.length/2-1])/2.0;
        }else {
            result = (double)arr[arr.length/2];
        }
        return result;
    }
}