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

最新推荐文章于 2023-08-01 09:45:00 发布

前路漫漫，继续前行！

最新推荐文章于 2023-08-01 09:45:00 发布

阅读量1.8k

点赞数

分类专栏： LeetCode算法刷题文章标签：算法

本文链接：https://blog.csdn.net/qq_40521306/article/details/108891132

版权

LeetCode算法刷题专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文探讨了一道寻找两个正序数组中位数的算法问题，要求设计时间复杂度为O(log(m+n))的解决方案。示例展示了不同输入情况下如何计算中位数，并提供了一个初步的解题思路，尽管时间复杂度较高，但表示了作者的独立思考过程。后续将优化代码以提高效率。

摘要由CSDN通过智能技术生成

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

题目

给定两个大小为 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

提示：

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

以下的代码是通过自己的思考得到的解决方案，虽然时间用了15ms,哈哈，用的时间较长，但自己也是比较高兴的，毕竟是自己独立思考的结果，后面会进一步优化代码，得到这道题更好的解决方案！！！

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int i=0;
        int j=0;
        int m=nums1.length+nums2.length;
        int[] result = new int[2];
        int curr = 0;
        int count = 0;
        int index = 0;   
        int num = 0;
        if(m%2 == 0){
            index = m/2;
            num = 2;
        }else{
            index = m/2+1;
            num = 1;
        }
        while(true){
            if(i==nums1.length){
                break;
            }
            if(j==nums2.length){
                break;
            }
            if(nums1[i]<nums2[j]){
                curr = nums1[i];
                i++;
            }else{
                curr = nums2[j];
                j++;
            }
            count++;
            if(count>index+1){
                break;
            }
            if((count == index || count == index+1) && num != 0){
                result[num-1] = curr;
                num--;
            }  
        }
        while(i<nums1.length){
            curr = nums1[i];
            i++;
            count++;
            if(count>index+1){
                break;
            }
            if((count == index || count == index+1) && num != 0){
                result[num-1] = curr;
                num--;
            }  
        }
        while(j<nums2.length){
            curr = nums2[j];
            j++;
            count++;
            if(count>index+1){
                break;
            }
            if((count == index || count == index+1) && num != 0){
                result[num-1] = curr;
                num--;
            }  
        }
        //System.out.println(result[0]);
        //System.out.println(result[1]);
        if(m%2==1){
            return result[0]+result[1];
        }else{
            return (result[0]+result[1])/2.0;
        }

    }
}