leetcode 第4题

最新推荐文章于 2022-03-29 13:39:05 发布

保尔柯察杨

最新推荐文章于 2022-03-29 13:39:05 发布

阅读量728

点赞数

分类专栏： LeetCode和牛客网

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.csdn.net/qq_34770510/article/details/78813964

版权

LeetCode和牛客网专栏收录该内容

27 篇文章 0 订阅

订阅专栏

leetcode 第4题： Median of Two Sorted Arrays

源码地址（github）： https://github.com/cy-hello/leedcode_004

一开始想到方法：
1 把2个数组合并成一个数组
2 然后按照从大到小或者从小到大的顺序排序
3 分类：判断合并后的数组的元素个数是奇数还是偶数
4 如果是奇数：取中间值；如果是偶数取中间2个数的平均值

class Solution {  
public:  
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2){  
        vector<int> nums;  
        if (nums1.size() > nums2.size()){  
            nums = nums1;  
            for (unsigned int i = 0; i < nums2.size(); i++){  
                nums.push_back(nums2[i]);  
            }  
        }  
        else{  
            nums = nums2;  
            for (unsigned int i = 0; i < nums1.size(); i++){  
                nums.push_back(nums1[i]);  
            }  
        }  
		
        sort(nums.begin(), nums.end());  
        if (nums.size() & 0x1){  
            return nums[nums.size() / 2]; 
        }  
        else{  
            return (nums[nums.size() / 2] + nums[(nums.size() / 2) - 1]) / 2.0;  
        }  
    }  
};

但是题目要求的时间复杂度是O(log(m+n))，虽然能通过，不符合题目要求，参考书上的代码

理解的不是很深刻。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        const int m = nums1.size();		/*取数组1的长度*/
        const int n = nums2.size();		/*取数组2的长度*/
        int total = m + n;
        if (total & 0x1)               /*数组总数是偶数*/
            return find_kth(nums1.begin(), m, nums2.begin(), n, total / 2 + 1);
        else                           /*数组总数是奇数*/
            return (find_kth(nums1.begin(), m, nums2.begin(), n, total / 2)+ find_kth(nums1.begin(), m, nums2.begin(), n, total / 2 + 1)) / 2.0;
    }
    private:
        static int find_kth(std::vector<int>::const_iterator nums1, int m,
        std::vector<int>::const_iterator nums2, int n, int k) {
        //always assume that m is equal or smaller than n
        if (m > n) return find_kth(nums2, n, nums1, m, k);
        if (m == 0) return *(nums2 + k - 1);
        if (k == 1) return min(*nums1, *nums2);
        //divide k into two parts
        int ia = min(k / 2, m), ib = k - ia;
        if (*(nums1 + ia - 1) < *(nums2 + ib - 1))
            return find_kth(nums1 + ia, m - ia, nums2, n, k - ia);
        else if (*(nums1 + ia - 1) > *(nums2 + ib - 1))
            return find_kth(nums1, m, nums2 + ib, n - ib, k - ib);
        else
            return nums1[ia - 1];
    }
};