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

题目: 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。 请你找出这两个有序数组的中位数
要求:时间复杂度为 O(log(m + n))。 你可以假设 nums1 和 nums2 不会同时为空。
示例: nums1 = [1, 3] nums2 = [2]
则中位数是 2.0

参考文章: 这里

进入递归之后就不用考虑奇偶了

	public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1 = nums1.length, len2 = nums2.length;
        int len = len1 + len2;
        // 只有len是偶数时 preK 和 k 才会不一样
        int preK = (len + 1)/ 2; 
        int k = (len + 2) / 2;  

        if ((len) % 2 == 0)
            return (helper(nums1, 0, len1 - 1, nums2, 0, len2 - 1, preK) 
                  + helper(nums1, 0, len1 - 1, nums2, 0, len2 - 1, k)) * 0.5;
        else 
            return helper(nums1, 0, len1 - 1, nums2, 0, len2 - 1, k);// preK 也可以
    }
   int helper(int[] nums1, int start1, int end1, int[] nums2, int start2, int end2, int k) {
        int len1 = end1 - start1 + 1;
        int len2 = end2 - start2 + 1;
        // 保证len1长度小于len2   
        if (len1 > len2) return helper(nums2, start2, end2, nums1, start1, end1, k); 

        // nums 从start2开始的第k个数(k从1开始 所以索引要-1) 
        if (len1 == 0) return nums2[start2 + k - 1];                      
        if (k == 1) return Math.min(nums1[start1], nums2[start2]); // 递归出口 

        // i 是 nums1 从start1 出发 第 k/2个数(不超过len1的情况下)
        int i = start1 + Math.min(len1, k / 2) - 1; // k / 2跟 len 比 防止越界
        int j = start2 + Math.min(len2, k / 2) - 1;

        if (nums1[i] < nums2[j])//nums1下一次比较开始索引前进i+1, i-start1+1: 这一次去掉的个数
            return helper(nums1, i + 1, end1, nums2, start2, end2, k - (i - start1 + 1)); 
        else 
            return helper(nums1, start1, end1, nums2, j + 1, end2, k - (j - start2 + 1));
            
   }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值