求两个有序数组的中位数

转自 success041000


求两个有序数组的中位数  

         如果有两个有序的数组,都是已经排好序的。那么求它们的中位数应该怎样求呢。如果采用对这两个数组进行排序的方法,最快的时间复杂度也要o(nlogn)的时间。但是,如果采用中位数和顺序统学的方法来寻找,则可以在o(n)的时间内解决这个问题。

       我们先寻找每个数组的中位数,因为是排好顺序的数组,因此,可以在o(1)时间内找到。然后,比较这两个数字的大小。如果A的中位数大于B的中位数,则在A的前半个数组和B的后半个数组中寻找,反之,在B的前半个数组和A的后半个数组寻找。根据递归方程,解得时间复杂度是o(n)。

       中位数:一组数据中间位置的数,如果是偶数个数,则取中间两个位置数的平均值。

       此题两个数组个数一样,那么两个数组的中位数总个数是2*n也就是偶数,中位数一定是中间两个位置的平均数。时间复杂度要求O(logn),则一定要充分利用数组有序的信息。

       网上看了很多版本,都是对奇偶数的考虑不全。自己试着编写了一下,发现很多细节很容易忽略。测试考虑的是整数,可以把数组直接全部设置为double类型。如果是整型,要考虑结果类型的强制转换。还有最后递归结束条件只考虑两数组中只剩下一个元素,或者中位数相等的两种情况。大家可以多找几个例子试验下。此代码在VC++6.0上测试过。大家可以自己再测试下,有问题的话欢迎提出。代码如下:

[cpp]  view plain copy
  1. //最后返回结果一定要是double,因为是两个中位数的平均值不一定是整数  
  2. double MidNum(int *A,int l1,int r1,int *B,int l2,int r2)  
  3. {  
  4.     //根据奇偶决定中位数的位置,要保证两个字数组元素个数相等  
  5.     int mid1,mid2;  
  6.     if( (r1-l1+1)%2==0 )    //偶数时  
  7.     {  
  8.         mid1=(l1+r1)/2+1;   //A取下中位数  
  9.         mid2=(l2+r2)/2;     //B取上中位数  
  10.     }  
  11.     else                    //奇数时  
  12.     {  
  13.         mid1=(l1+r1)/2;  
  14.         mid2=(l2+r2)/2;  
  15.     }  
  16.   
  17.     if(l1==r1 && l2==r2)            //最后两个数组都剩下一个元素  
  18.         return (double)(A[l1]+B[l2])/2;       
  19.     //最后两个数组都剩下两个元素,而且A[mid1]>B[mid2],底下的情况处理不了,一直递归下去  
  20.     //如A[6]={1,3,5,6,8,10};B[6]={2,4,7,9,11,15};最后A{6,8},B{4,7}  
  21.     if( r1-l1==1 && r2-l2==1 )        
  22.     {  
  23.         if(A[mid1]>B[mid2])          //得对剩下的4个数排序A[l1],A[r1](r1==mid1),B[l2](l2==mid2),B[r2]  
  24.         {  
  25.             if(B[r2]<=A[l1])     //B[mid2],B[r2],A[l1],A[mid1]  
  26.                 return (double)(B[r2]+A[l1])/2;  
  27.             else if(A[l1]<=B[mid2] && A[mid1]>=B[r2]) //A[l1],B[mid2],A[mid1],B[r2]  
  28.                 return (double)(B[mid2]+A[mid1])/2;  
  29.             else if(A[l1]<=B[mid2] && A[mid1]<B[r2])  //A[l1],B[mid2],B[r2],A[mid1]  
  30.                 return (double)(B[mid2]+B[r2])/2;             
  31.         }  
  32.     }  
  33.     if(A[mid1]==B[mid2])  
  34.         return A[mid1];  
  35.     else if( A[mid1] > B[mid2])  
  36.         return MidNum(A,l1,mid1,B,mid2,r2);  
  37.     else   
  38.         return MidNum(A,mid1,r1,B,l2,mid2);  
  39. }  
  40.   
  41.   
  42. int main()  
  43. {  
  44.     int A[6]={1,3,5,6,8,10};  
  45.     int B[6]={2,4,6,9,11,15};  
  46.   
  47.     double m2=MidNum(A,0,5,B,0,5);  
  48.     cout<<m2<<endl;  
  49.   
  50.     int A2[10]={17,18,28,37,42,54,63,72,89,96};  
  51.     int B2[10]={3,51,71,72,91,111,121,131,141,1000};  
  52.   
  53.     double m=MidNum(A2,0,9,B2,0,9);  
  54.     cout<<m<<endl;  
  55.   
  56.     return 0;  
  57. }  


  • 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、付费专栏及课程。

余额充值