每日一题（day4）

最新推荐文章于 2023-11-27 20:47:27 发布

Gabriel_Li

最新推荐文章于 2023-11-27 20:47:27 发布

阅读量405

点赞数

分类专栏：每日一题文章标签： leetcode 递归

本文链接：https://blog.csdn.net/Gabriel_Li/article/details/51870961

版权

每日一题专栏收录该内容

5 篇文章 0 订阅

订阅专栏

Leetcode004

Median of Two Sorted Arrays问题
知识更新：求k小值、递归

private:
    double findKth(vector<int>& nums1, int i, vector<int>& nums2, int j, int k)
    {
        int len1 = nums1.size() - i;        
        int len2 = nums2.size() - j;
        if (len1 > len2) return findKth(nums2, j, nums1, i, k);
        if (!len1) return nums2[k - 1];
        if (k == 1) return min(nums1[i], nums2[j]);     
        int cur1, cur2;
        len1 < k / 2 ? cur1 = i + len1 - 1 : cur1 = i + k / 2 - 1;
        cur2 = j + k - (cur1 - i + 1) - 1;
        if (nums1[cur1] < nums2[cur2]) 
            return findKth(nums1, cur1 + 1, nums2, j, k - (cur1 - i + 1));
        if (nums1[cur1] > nums2[cur2])
            return findKth(nums1, i, nums2, cur2 + 1, k - (cur2 - j + 1));
        return nums1[cur1];
    }
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) { 
        int total_len = nums1.size() + nums2.size();
        if (total_len & 1) return findKth(nums1, 0, nums2, 0, total_len / 2 + 1);
        else return (findKth(nums1, 0, nums2, 0, total_len / 2 + 1) 
            + findKth(nums1, 0, nums2, 0, total_len / 2)) / 2;
    }