两个排序数组的中位数

求两个排序数组中位数,这道题是很有意思的一道题目,算法导论中9.3-8题,这题必须在O(logn)的时间复杂度求解,否则肯定悲剧。。。

这题有个关键的条件,那就是这两个数组长度相等

思路如下:

数组A:1, 3, 5, 7, 9

数组B:2, 4, 6, 8, 10

首先取二者的中位数,在O(1)时间复杂度内求出,那么数组A的midValue = 5,数组B的midValue = 6,则较小的元素midValue的左边所有元素,也就是midPosition个,和较大元素右边的所有元素,都要去掉,由于去掉的元素占所有元素的一 半,所以复杂度为O(logn)。只需要移动begin和end的position即可。

第一轮:

数组A:1, 3, 5, 7, 9

数组B:2, 4, 6, 8, 10

第二轮:

数组A:1, 3, 5, 7, 9

数组B:2, 4, 6, 8, 10

只剩下两个元素,由于这种情形下midValue始终是偏左的元素,因此循环无法退出,所以满足lhsBegin == lhsEnd - 1和 rhsBegin == rhsEnd - 1时,循环退出!

如果两个

上述情形是当数组中有两个及以上元素才成立,必须考虑两个数组只存在一个元素的情形,那么直接取较小的元素即可


    #include <iostream>  
      
    using namespace std;  
      
    int findMidValue(int *lhs, int *rhs, int size)  
    {  
        if (lhs == NULL || rhs == NULL || size <= 0)  
            return -1;  
      
        int lhsBegin = 0;  
        int rhsBegin = 0;  
        int lhsEnd = size - 1;  
        int rhsEnd = size - 1;  
        int result;  
      
        while ((lhsBegin < lhsEnd && rhsBegin < rhsEnd))  
        {  
            if ((lhsBegin == lhsEnd - 1 && rhsBegin == rhsEnd - 1))  
                break;  
            int lhsMid = (lhsBegin + lhsEnd) >> 1;  
            int rhsMid = (rhsBegin + rhsEnd) >> 1;  
              
            if (lhs[lhsMid] == rhs[rhsMid])  
            {  
                result = lhs[lhsMid];  
                break;  
            }  
            else if (lhs[lhsMid] < rhs[rhsMid])  
            {  
                lhsBegin = lhsMid;  
                rhsEnd = rhsMid;  
            } else  
            {  
                lhsEnd = lhsMid;  
                rhsBegin = rhsMid;  
            }  
        }  
      
        if (lhsBegin == lhsEnd && rhsBegin == rhsEnd)  
            result = min(lhs[lhsBegin], rhs[rhsBegin]);  
        else if (lhsBegin == lhsEnd - 1 && rhsBegin == rhsEnd - 1)  
        {  
            if (lhs[lhsBegin] < rhs[rhsBegin])  
            {  
                result = min(lhs[lhsEnd], rhs[rhsBegin]);  
            }  
            else  
            {  
                result = min(lhs[lhsBegin], rhs[rhsEnd]);  
            }  
        }  
      
        return result;  
    }  
      
    void main()  
    {  
        int lhs[] = {1, 2, 3, 4};  
        int rhs[] = {0, 2, 3, 4};  
        const int size = sizeof lhs / sizeof *lhs;  
        int result = findMidValue(lhs, rhs, size);  
        cout << "mid value = " << result << endl;  
    }  


     

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
假设有两个有序数组 nums1 和 nums2,长度分别为 m 和 n。要找到这两个数组中位数,时间复杂度要为 O(log(m+n))。 一种思路是利用归并排序的思想,将两个有序数组合并成一个有序数组,然后找到中位数。 具体步骤如下: 1. 初始化两个指针 i 和 j,分别指向 nums1 和 nums2 的起始位置。 2. 判断两个指针所指元素的大小关系,将较小的元素加入到一个新的数组中,并将指针向后移动一位。 3. 重复步骤 2,直到任意一个指针越界。 4. 判断剩余数组的长度,将剩余的元素加入到新的数组中。 5. 找到新数组中位数,如果新数组长度为偶数,则取中间两个数的平均值,如果长度为奇数,则取中间的数。 代码实现如下: ``` double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) { int m = nums1.size(), n = nums2.size(); int total = m + n; vector<int> nums(total); int i = 0, j = 0, k = 0; while (i < m && j < n) { if (nums1[i] < nums2[j]) { nums[k++] = nums1[i++]; } else { nums[k++] = nums2[j++]; } } while (i < m) { nums[k++] = nums1[i++]; } while (j < n) { nums[k++] = nums2[j++]; } if (total % 2 == 0) { return (nums[total / 2 - 1] + nums[total / 2]) / 2.0; } else { return nums[total / 2]; } } ``` 时间复杂度为 O(m+n),不符合题目要。可以使用二分查找的方法将时间复杂度优化到 O(log(m+n))。 具体思路如下: 1. 假设两个有序数组的长度分别为 m 和 n,将 nums1 分为两部分,前一部分包含 i 个元素,后一部分包含 m-i 个元素;将 nums2 分为两部分,前一部分包含 j 个元素,后一部分包含 n-j 个元素。 2. 如果中位数两个数组的左半部分,那么 i 和 j 都需要向右移动;如果在右半部分,i 和 j 都需要向左移动;如果 i 和 j 恰好满足条件,则找到了中位数。 3. 不断重复步骤 2,直到找到中位数。 代码实现如下: ``` double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) { int m = nums1.size(), n = nums2.size(); if (m > n) { swap(nums1, nums2); swap(m, n); } int left = 0, right = m, halfLen = (m + n + 1) / 2; while (left <= right) { int i = (left + right) / 2; int j = halfLen - i; if (i < m && nums2[j-1] > nums1[i]) { left = i + 1; } else if (i > 0 && nums1[i-1] > nums2[j]) { right = i - 1; } else { int maxLeft = 0; if (i == 0) { maxLeft = nums2[j-1]; } else if (j == 0) { maxLeft = nums1[i-1]; } else { maxLeft = max(nums1[i-1], nums2[j-1]); } if ((m + n) % 2 == 1) { return maxLeft; } int minRight = 0; if (i == m) { minRight = nums2[j]; } else if (j == n) { minRight = nums1[i]; } else { minRight = min(nums1[i], nums2[j]); } return (maxLeft + minRight) / 2.0; } } return 0.0; } ``` 时间复杂度为 O(log(min(m,n)))。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值