尋找兩個有序陣列的中位數

呼勺要b3發表於2019-05-14

給定兩個大小為 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
public class Solution {  public double findMedianSortedArrays(int[] nums1, int[] nums2) {    int cur1 = 0;    int cur2 = 0;    int len1 = getArrayLength(nums1);    int len2 = getArrayLength(nums2);    int res = 0;    for (int i = 0; i < (len1 + len2 + 1) >> 1; i++) {      if (cur1 >= len1 || cur2 < len2 && nums1[cur1] > nums2[cur2]) {
        res = nums2[cur2++];
      } else {
        res = nums1[cur1++];
      }
    }    if (((len1 + len2) & 1) != 0) {      return (double) res;
    } else {      if (cur1 >= len1) {        return ((double) res + (double) nums2[cur2]) / 2;
      }      if (cur2 >= len2) {        return ((double) res + (double) nums1[cur1]) / 2;
      }      return ((double) res + (double) Math.min(nums1[cur1], nums2[cur2])) / 2;
    }
  }  private int getArrayLength(int[] arr) {    return arr == null ? 0 : arr.length;
  }
}


來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31551894/viewspace-2644400/,如需轉載,請註明出處,否則將追究法律責任。

相關文章