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

题目

给定两个大小为 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  double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int nums1Length = nums1.length;
    int nums2Length = nums2.length;

	//移动的次数,偶数减1
    int centre = (nums1Length + nums2Length) % 2 == 0 ? (nums1Length + nums2Length) / 2 - 1 : (nums1Length + nums2Length) / 2;

    int nums1Left = 0;
    int nums2Left = 0;
    while(centre > 0) {
        int value1 = nums1Left == nums1Length ? Integer.MAX_VALUE : nums1[nums1Left];
        int value2 = nums2Left == nums2Length ? Integer.MAX_VALUE : nums2[nums2Left];
        if (value1 < value2) {
            nums1Left++;
        } else {
            nums2Left++;
        }
        centre--;
    }

	
    int first = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
            ? nums2[nums2Left] : nums1[nums1Left];
    //两个数组长度和为偶数时
    if ((nums1Length + nums2Length) % 2 == 0) {
    	//如果nums1Left没有走到最后,并且first值等于nums1[nums1Left],表示first是在nums1数组
        if (nums1Left != nums1Length && first == nums1[nums1Left]) {
            nums1Left++;
        } else {
            nums2Left++;
        }
        int second = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
                ? nums2[nums2Left] : nums1[nums1Left];
        return (first + second) / 2.0;
    }
    return (double)first;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值