欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

两个排序数组的中位数

程序员文章站 2024-03-16 12:50:34
...

题目描述


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

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

你可以假设 nums1 和 nums2 不同时为空。

示例 1:

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

中位数是 2.0

示例 2:

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

中位数是 (2 + 3)/2 = 2.5

思路:先对两个子数组合并为一个数据,然后利用冒泡排序对合并后的数据进行排序。求中位数,如果是偶数个数,中位数就是中间两个数的和除以2;如果是奇数个数,中位数就是中间的那个数。

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    void bubble(int *a,int len);
    int len = nums1Size + nums2Size;
    int myArray[1000];
    int *p;
    double mid = -1.0;
    p = myArray;
    int i,j;
    for(i=0;i<nums1Size;i++)
    {
        *(p+i) = *nums1++;
    }
    for(i = nums1Size-1,j=0;i<nums1Size + nums2Size;i++,j++)
    {
        *(p+i) = *nums2;
    }
    bubble(*p,len);
    if(len%2==0) 
        mid = 1.0*(*(p+(len-1)/2 )+ *(p + (len-1)/2 + 1))/2;
    else 
        mid = 1.0*(*(p + (len-1)/2));            
                               
    return mid;
}

void bubble(int *a,int len)
{
    int i,j;
    int temp;
    int *pt;
    pt = a;
    for(i=0;i<len-1;i++)
    {
        for(j=0;j<len-1-i;j++)
        {
            if(*(pt+j)>*(pt+j+1))
            {
                temp = *(pt+j+1);
                *(pt+j+1) = *(pt+j);
                *(pt+j) = temp;
            }
            
        }
    }
}