力扣(LeetCode) 4.寻找两个正序数组的中位数 java

4.寻找两个正序数组的中位数(困难)

题目如下:

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

示例 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

提示：

• nums1.length == m
• nums2.length == n
• 0 <= m <= 1000
• 0 <= n <= 1000
• 1 <= m + n <= 2000
• -10⁶ <= nums1[i], nums2[i] <= 10⁶

方法一

这题我看了一会，愣是没感觉到它困难体现在哪里，感觉很好做啊，可能难在最优解不好想吧

一上来当然不可能暴力，虽然暴力代码很简单，但效率实在不可取，但可以说下暴力的思路，那就是把这两个数组合成一个，然后排序，再取中位数即可，简单明了，但我们毕竟已经写到第4题了，早已不是吴下阿蒙了，所以要相处更有效率的解法。

不需要合并两个有序数组，只要找到中位数的位置即可。由于两个数组的长度已知，因此中位数对应的两个数组的下标之和也是已知的。维护两个指针，初始时分别指向两个数组的下标 0 的位置，每次将指向较小值的指针后移一位（如果一个指针已经到达数组末尾，则只需要移动另一个数组的指针），直到到达中位数的位置。

上面这段是力扣的题解。我日常结束题目后去看题解，发现我这个解法只是顺带一提(哭泣)，最优解先不提，我发现我这个解法不管怎么解释感觉都没上面说的清楚，所以拷下来直接放上去了

代码如下，注释详细

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        //先获取两个数组的长度
        int m = nums1.length;
        int n = nums2.length;
        int count = 0;
        //注意这个中位数，
        //如果两个数组长度和sum为偶数，那么中位数是 sum / 2 以及 sum / 2 + 1  这两个数
        //比如长度和为4时，中位数就是(第2个数+第3个数) / 2  这个上面示例2就是如此
        //如果长度和sum为计数，那么中位数就是sum / 2 + 1，比如长度和为5时，中位数就是(第3个数) / 2
        int mid = (m + n) / 2;
        if((m + n) % 2 == 1){
            mid++;
        }
        //i j就是对应的两个指针
        int i = 0;
        int j = 0;
        //中位数 因为中位可能是两个数的和
        int sum = 0;
        for(int k = 0;k < 1001;k++){
            //计数 用它来判断是不是到中位了
            count++;
            
            //分别把当前两个数组的数取出来 如果一个数组已经结束了 那么就取1000001 题中已经给出数组中值不会大于10的6次
            int num1 = 1000001;
            if(i < m){
                num1 = nums1[i];
            }

            int num2 = 1000001;
            if(j < n){
                num2 = nums2[j];
            }

            //根据这两个数组的值来移动指针
            if(num1 <= num2){
                i++;
            }else{
                j++;
            }

            //如果到达中位数
            if(count == mid || count == mid + 1){
                sum += num1 <= num2 ? num1 : num2;
                //如果总数组长为奇数 直接返回
                if((m + n) % 2 == 1){
                    return sum * 1.0;
                }
                
                //如果总数组长为偶数 要取到第mid + 1时再结束
                if((m + n) % 2 == 0 && count == mid + 1){
                    return sum / 2.0;
                }
            }
        }

        return 0.0;
    }
}

纯暴力的时间复杂度是 O(m+n)，空间复杂度是 O(m+n)。上面这种双指针法虽然可以将空间复杂度降到O(1)，但是时间复杂度仍是 O(m+n)(其实也没有啦，只找中位数O((m+n) / 2)就够了)。

写完一执行，时间执行会了3ms，偶尔能上2ms。时间空间都打败8 90%左右，但显然不是最优解了。唏嘘

看了题解，有一个时间复杂度为O(log (m+n))的解法，用到了二分查找，有兴趣的可以去看下

，只找中位数O((m+n) / 2)就够了)。

写完一执行，时间执行会了3ms，偶尔能上2ms。时间空间都打败8 90%左右，但显然不是最优解了。唏嘘

看了题解，有一个时间复杂度为O(log (m+n))的解法，用到了二分查找，有兴趣的可以去看下