O(log(min(m,n)))时间复杂度 在两个有序数组中查找第k小的数

public class Solution {
/*
仿照https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
的思路写出
*/
public static int findTheKthNumber(int[] nums1, int[] nums2,int k) {
    if (nums1.length > nums2.length) {
        return findTheKthNumber(nums2, nums1,k);
    }

    int m = nums1.length;
    int n = nums2.length;
    int left = 0, right = m;
    // median1:前一部分的最大值
    // median2:后一部分的最小值
    int median1 = 0;
    int median2 = 0;

    while (left <= right) {
        // 前一部分包含 nums1[0 .. i-1] 和 nums2[0 .. j-1]
        // 后一部分包含 nums1[i .. m-1] 和 nums2[j .. n-1]
        int i = (left + right) / 2;
         int j = k-i;

        // nums1Left,nums1Right,nums2Left,nums2Right 分别表示 nums1[i-1], nums1[i], nums2[j-1], nums2[j]
        int nums1Left = (i == 0 ? Integer.MIN_VALUE : nums1[i - 1]);
        int nums1Right = (i == m ? Integer.MAX_VALUE : nums1[i]);
        int nums2Left = (j == 0 ? Integer.MIN_VALUE : nums2[j - 1]);
        int nums2Right = (j == n ? Integer.MAX_VALUE : nums2[j]);

        if (nums1Left <= nums2Right) {
            
            median1 = Math.max(nums1Left, nums2Left);
            median2 = Math.min(nums1Right, nums2Right);
            left = i + 1;
        }
        else {
            right = i - 1;
        }
    }

    return median1;
}



public static void main(String[] args) {
	int[]nums1= {0,1};
	int[]nums2= {2,3,4,5};
	System.out.println(findTheKthNumber(nums1,nums2,3));
}

}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值