LeetCode OJ-4.Median of Two Sorted Arrays-优快云博客

本文链接：https://blog.youkuaiyun.com/tobebetterprogrammer/article/details/70766959

本文解析了LeetCode上的第4题——两个有序数组的中位数问题，并提供了一种符合O(log(m+n))时间复杂度要求的解决方案。通过合并思想，文章详细介绍了如何找出两个有序数组的中位数。

LeetCode OJ-4.Median of Two Sorted Arrays

题目描述

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

Example 1:
nums1 = [1, 3]
nums2 = [2]

The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

题目理解

在两个有序的数组里，寻找中位数，要求时间复杂度为O(log(m + n))。这里使用归并的思想，将两个数组中前(m + n) / 2 + 1个元素放入另一个数组中，如果m + n为奇数，则中位数就是该数组中最后一个元素；如果m + n为偶数，则中位数就是倒数两个元素的平均值。

具体的代码如下：

Code

double find_median(vector<int> &v1, vector<int> &v2)
{
    size_t sz = (v1.size() + v2.size()) / 2 + 1;
    vector<int> v3(sz);
    int i, j, k;
    i = 0;
    j = 0;
    k = 0;
    size_t v1_sz = v1.size();
    size_t v2_sz = v2.size();
    while (k < sz) {
        if (i < v1_sz && j < v2_sz) {
            if (v1[i] <= v2[j]) {
                v3[k++] = v1[i++];
            }
            else {
                v3[k++] = v2[j++];
            }
        }
        else {
            if (i < v1_sz) {
                v3[k++] = v1[i++];
            }

            if (j < v2_sz) {
                v3[k++] = v2[j++];
            }
        }
    }

    double res = 0;
    if ((v1_sz + v2_sz) % 2 != 0) {
        res = v3[v3.size() - 1];
    }
    else {
        res = (v3[v3.size() - 1] + v3[v3.size() - 2]) / 2.0;
    }

    return res;
}