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

题目：寻找两个正序数组的中位数

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

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

示例 1：

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

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000

答案

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1=nums1.length,len2=nums2.length;
        int total=len1+len2;
        double ans;
        //如果是偶数，则中位数为中间那位数
        if (total%2==1){
            ans=getMedian(nums1,nums2,total/2+1);
         //如果是奇数，则中位数为中间两个数的平均数，注意采用double类型且/2.0,/2会得到整数
        }else {
            ans=(getMedian(nums1,nums2,total/2)+getMedian(nums1,nums2,total/2+1))/2.0;
        }
        return ans;
    }

    //k用于求两个数组按顺序排之后下表为k的那位数
    public int getMedian(int[] num1,int[] num2,int k){
        //n1和n2分别记录num1和num2当前被排除的值
        // 如：n1=2,表示num1[0]到num1[1]都被排除，将num1[2]作为数组第一个元素
        int n1=0,n2=0;

        //满足边界条件退出，否则一直循环
        while (true){
         //当n1==num1.length时，n1的所有数都被排除
         //只需要从num2中从当前有效数开始数k个数
        if (n1==num1.length){
            return num2[n2+k-1];
        }
        //当n1==num1.length时，n1的所有数都被排除
        if (n2==num2.length){
            return num1[n1+k-1];
        }
        //当k==1时，为下一个数，按顺序排，小的数在前，即为所找的数
        if (k==1){
            return Math.min(num1[n1],num2[n2]);
        }

        //为了防止下标溢出，当num1和num2长度相差大时容易发生下标溢出
        int newn1 =Math.min(n1+k/2,num1.length)-1;
        int newn2 =Math.min(n2+k/2,num2.length)-1;

        //假设两个有序数组分别是 A和 B。
        // 要找到第k个元素，我们可以比较 A[k/2−1]和 B[k/2−1]。
        // 由于 A[k/2−1]和 B[k/2−1]的前面都有k/2-1个数，
        // 对于 A[k/2−1]和 B[k/2−1]中的较小值，最多也只会有k−2(k/2-1+k/2-1)个元素比它小，那么它就不是第k小的数。
        if (num1[newn1]<=num2[newn2]){
            k=k-(newn1-n1+1);
            n1 = newn1+1;
        }else {
            k=k-(newn2-n2+1);
            n2 = newn2+1;
        }

        }
    }
}