两个排序数组的中位数-合并排序取中间值法

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n))。

示例 1:

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

中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

中位数是 (2 + 3)/2 = 2.5
public double FindMedianSortedArrays(int[] nums1, int[] nums2) {
    int m = nums1.Length;
    int n = nums2.Length;
    int i = 0;
    int j = 0;
    int index = 0;
    int[] datas = new int[m+n];

    //合并排序
    while(i<m && j<n)
    {
        if(nums1[i] < nums2[j])
        {
            datas[index++] = nums1[i++];
        }
        else
        {
            datas[index++] = nums2[j++];
        }
    }

    //填充剩下的
    while(j<n)
    {
        datas[index++] = nums2[j++];
    }

    while(i<m)
    {
        datas[index++] = nums1[i++];
    }

    int mid = (m+n)/2;
    int offset = (m+n)%2;
    if(offset == 0)
    {
        return (datas[mid]+datas[mid-1])/2.0f;
    }else
    {
        return datas[mid];
    }
}


使用合并排序的方式,将两个数组合并排序到一个数组中,然后判断长度为偶数就取中间两个的平均数,为奇数就去取中间那位。时间复杂度应该是O(n)或O(m),空间复杂度是O(m+n)。还有更优的解法。


首页 我的博客
粤ICP备17103704号