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));
}
}