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

Am最温柔

于 2022-05-02 21:13:58 发布

阅读量331

点赞数

分类专栏：菜菜做题文章标签： c++ leetcode

本文链接：https://blog.csdn.net/weixin_43919570/article/details/124546673

版权

二分查找有序数组中位数算法排序

关键词由CSDN通过智能技术生成

菜菜做题专栏收录该内容

91 篇文章 0 订阅

订阅专栏

可以理解为寻找第k个数字。
使用二分的思想，将两个数组的第k/2个数字进行比较，一次排除k/2个数字。
如果数组长度小于k/2需要取数组剩余长度，如果一个数组已经处理到最后了，那么直接返回另一个数组的给定位置的值即可。
getKth中的k可以理解为找第k个数字（从1开始，找第1个，第2个，…，第k个数字）。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size();
        int len2 = nums2.size();
        int k = (len1 + len2) / 2;
        if (0 == (len1 + len2) % 2) {
            return (getKth(nums1, 0, len1-1, nums2, 0 ,len2-1, k) + getKth(nums1, 0, len1-1, nums2, 0 ,len2-1, k+1)) / 2;
        } else {
            return getKth(nums1, 0, len1-1, nums2, 0 ,len2-1, k+1);
        }
        return 0;
    }
    double getKth(vector<int>& nums1, int nums1Beg, int nums1End, vector<int>&nums2, 
            int nums2Beg, int nums2End, int k) {
        // 如果长度不够，需要特殊处理
        if (nums1Beg > nums1End) {
            return nums2[nums2Beg + k -1];
        } else if (nums2Beg > nums2End) {
            return nums1[nums1Beg + k -1];
        }
        
        // 0.如果k==1，那么谁小返回谁
        if (1 == k) {
            return nums1[nums1Beg] < nums2[nums2Beg] ? nums1[nums1Beg] : nums2[nums2Beg];
        }
        
        // 如果nums1长度不够
        int minLen1 = k/2 > nums1End - nums1Beg + 1 ? nums1End - nums1Beg + 1 : k/2; 
        // 如果nums2长度不够
        int minLen2 = k/2 > nums2End - nums2Beg + 1 ? nums2End - nums2Beg + 1 : k/2; 
        // 1.取k/2，比大小，谁小排除谁,找第k-k/2个，递归
        if (nums1[nums1Beg + minLen1 - 1] < nums2[nums2Beg + minLen2 - 1]) {
            return getKth (nums1, nums1Beg + minLen1, nums1End, nums2, nums2Beg, nums2End, k - minLen1);
        } else if (nums1[nums1Beg + minLen1 - 1] >= nums2[nums2Beg + minLen2 - 1]) {
            return getKth(nums1, nums1Beg, nums1End, nums2, nums2Beg + minLen2, nums2End, k - minLen2);
        }
        return 0;
    }
};