ITPub博客

首页 > 大数据 > 数据分析 > 寻找两个有序数组的中位数

寻找两个有序数组的中位数

原创 数据分析 作者:呼勺要b3 时间:2019-05-14 22:46:53 0 删除 编辑

给定两个大小为 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博客 ” ,链接:https://blog.itpub.net/31551894/viewspace-2644400/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2019-04-17

  • 博文量
    7
  • 访问量
    8626