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

给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1:

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

解题代码:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
      int m = nums1.length;
      int n = nums2.length;
      int[] nums3 = new int[m+n];
      if (m == 0){
        if (n % 2 == 0){
          return (nums2[n/2] + nums2[n/2-1])/2.0;
        }else {
          return nums2[(n-1)/2];
        }
      }
      if (n == 0){
        if (m % 2 == 0){
          return (nums1[m/2] + nums1[m/2-1])/2.0;
        }else {
          return nums1[(m-1)/2];
        }
      }
      // 当m,n都大于0时
      int j = 0, k = 0;
      for (int i = 0; i < m+n; i++) {
        if (j < m && k < n){
          if (nums1[j] <= nums2[k]){
            nums3[i] = nums1[j++];
            continue;
          }else {
            nums3[i] = nums2[k++];
            continue;
          }
        }
        if (j >= m && k < n){
          nums3[i] = nums2[k++];
          continue;
        }
        if (k >= n && j < m){
          nums3[i] = nums1[j++];
          continue;
        }
      }
      if ((m+n) % 2 == 0){
        return (nums3[(m+n)/2] + nums3[(m+n)/2-1])/2.0;
      }else {
        return nums3[(m+n-1)/2];
      }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值