題目描述

給定兩個大小分別為 m 和 n 的正序（從小到大）陣列 nums1 和 nums2。請你找出並返回這兩個正序陣列的 中位數 。

演算法的時間複雜度應該為 O(log (m+n)) 。

示例

輸入：nums1 = [1,3], nums2 = [2]

輸出：2.00000

解釋：合併陣列 = [1,2,3] ，中位數 2

題目分析

leetcode上標註的是一個困難題，但是做起來感覺應該是一個簡單題。題目中需要找到兩個陣列的中位數。首先，如果資料的個數為雙數，則中位數是處於中間的兩個數的平均值，否則就是中間那個數。然後，兩個陣列都是有序的，我們只需要將兩個陣列按順序遍歷一遍，找到中位數即可

解法1

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int size = nums1.length + nums2.length;
    int start = size % 2 == 0 ? size / 2 + 1 : size / 2 + 2;
    int i = 0 , j = 0;
    int index = 0;
    int current = 0;
    int last = 0;
    while (i < nums1.length && j < nums2.length && index < start) {
        
 
++ index;
        last = current;
        if (nums1[i] < nums2[j]) {
            current = nums1[i];
            ++ i;
        } else {
            current = nums2[j];
            ++j;
        }
    }
    while (i < nums1.length && index < start) {
        ++ index;
        last = current;
        current 
 
= nums1[i];
        ++ i;
    }
    while (j < nums2.length && index < start) {
        ++ index;
        last = current;
        current = nums2[j];
        ++ j;
    }
    if (index == start) {
        if (size % 2 == 0) {
           return (double) (last + current) / 2;
        }
        return last;
    }
    return current;
}