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

JehanRio

已于 2023-09-04 09:52:39 修改

阅读量146

点赞数 1

分类专栏： Leetcode专栏文章标签： leetcode 算法职场和发展

于 2023-07-02 11:22:18 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/131499039

版权

Leetcode专栏专栏收录该内容

278 篇文章 2 订阅

订阅专栏

文章目录

题目
代码（首刷自解，复杂度m+n)
代码（首刷看解析，复杂度log(m+n)）
代码（9.4 二刷看解析）

题目

在这里插入图片描述
Leetcode 4. 寻找两个正序数组的中位数

代码（首刷自解，复杂度m+n)

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int p1 = nums1.size(), p2 = nums2.size();
        int i = 0, j = 0;
        vector<int> nums3(p1+p2+2);
        int k3 = 0;
        while(i < p1 && j < p2) {
            nums3[k3++] = nums1[i] < nums2[j] ? nums1[i++] : nums2[j++];
        }
        while(i < p1) {
            nums3[k3++] = nums1[i++];
        }
        while(j < p2) {
            nums3[k3++] = nums2[j++];
        }
        return k3 % 2 ? nums3[k3/2] : (nums3[k3/2-1] + nums3[k3/2]) / 2.0;
    }
};

代码（首刷看解析，复杂度log(m+n)）

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int total = nums1.size() + nums2.size();
        if(total % 2) {
            return getKthElement(nums1, nums2, (total + 1) / 2);
        } else {
            return (getKthElement(nums1, nums2, total / 2) + getKthElement(nums1, nums2, total / 2 + 1)) / 2.0;
        }
    }
    int getKthElement(vector<int>& nums1, vector<int>& nums2, int k) {
        int m = nums1.size(), n = nums2.size();
        int i = 0, j = 0;
        while(1) {
            // 边界情况
            if(i == m) 
                return nums2[j+k-1];
            if(j == n)
                return nums1[i+k-1];
            if(k == 1)
                return min(nums1[i], nums2[j]);

            // 正常情况
            int newIndex1 = min(i + k / 2 - 1, m - 1);
            int newIndex2 = min(j + k / 2 - 1, n - 1);
            int pivot1 = nums1[newIndex1];
            int pivot2 = nums2[newIndex2];
            if(pivot1 <= pivot2) {
                k -= newIndex1 - i + 1;
                i = newIndex1 + 1;
            } else {
                k -= newIndex2 - j + 1;
                j = newIndex2 + 1;
            }
        }
    }
};

代码（9.4 二刷看解析）

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size(), n = nums2.size();
        int k = m+n;
        if(k%2) {
            return getKthElement(nums1, nums2, (k+1)/2);
        }
        return (getKthElement(nums1, nums2, k/2) + getKthElement(nums1, nums2, k/2+1)) / 2.0;
    }   
    int getKthElement(vector<int>& nums1, vector<int>& nums2, int k) {
        int m = nums1.size(), n = nums2.size();
        int i = 0, j = 0;
        while(1) {
            if(i == m) 
                return nums2[j+k-1];
            if(j == n)
                return nums1[i+k-1];
            if(k == 1)
                return min(nums1[i], nums2[j]);
                
            int next1Key = min(m-1, i+k/2-1);
            int next2Key = min(n-1, j+k/2-1);
            int pivot1 = nums1[next1Key];
            int pivot2 = nums2[next2Key];
            if(pivot1 <= pivot2) {
                k -= next1Key - i + 1;
                i = next1Key + 1;
            } else {
                k -= next2Key - j + 1;
                j = next2Key + 1;
            }
        }
    }
};