leetcode刷题记录2021年3月19日

array-0004-Median of Two Sorted Arrays

题目描述

寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5

示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000

示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000

示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000

范围:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-10^6 <= nums1[i], nums2[i] <= 10^6

思路有两种,暂时只研究了其中一种,另一种等等再说。

思路

思路是leetcode cookbook这本书上的。

寻找中位数的问题可以看成寻找把数组划成两部分的那条线,线的旁边就是中位数,因为现在是寻找两个数组合并之后的中位数,如果先合并在找铁定时间不够。
我们可以先将其中一个划分成两半,那么另一个数组的划分也就确定了,因为中位数的位置是固定的,所以只需要确定一个数组的划分方式,如何判断当前的划分是否正确呢,只要保证这两条线左边最大的值小于右边最小的值就行了,一个简单的比较即可。
假设其中一个划分是在num1数组的第n-1个数和第n个数之间和num2数组的第m-1的第m个数之间,那么,只要满足:

max(num1[n-1],num2[m-1])<min(num1[n],num2[m])

这种划分就是正确的,假设(n+m)%2==0,那么,中位数就是

(max(num1[n-1],num2[m-1])+min(num1[n],num2[m]))/2

反之,中位数是:

max(num1[n-1],num2[m-1])

因为是有序数组,所以上面的式子可以可以简化成:

num1[n-1]<num2[m]&&num2[m-1]<num1[n]

因此,整个算法就可以围绕着其中一个数组的划分上,由于要降低时间复杂度,所以用二分搜索来实现划分的查找以及选择较小的数组进行查找

编程细节

虽然上面的思路并不复杂,但是实现的时候有很多需要注意的地方,尤其是我还是个初学者,我看的leetcode cookbook上的实现很多地方的细节都看不懂,因此在研究了一番之后,总结了一下理解这个实现所需要的一些前提。

二分算法

程序的主题就是二分算法的一个魔改
基础模板

void test_erfen()
{
    int p[50],l=0,r=50-1,n=50;
    srand(NULL);
    for(int i=0;i<1;i++)
        p[i]=i;
    int mid = (l+r)/2;
    while(l<=r)
    {
        mid=(l+r)/2;
        //mid=l+(r-l)>>1; 可能比较快
        if(p[mid]<49)
            l=mid+1;
        else if(p[mid]>49)
            r=mid-1;
        else
            break;
    }
}

二分查找最基础的几个变量有代表当前范围的范围的left,right,注意这是两边都闭合的区间,代表当前查找对象的mid,结束之后,mid就是结果,查找不到就是left>right,

中位数

数组长度为n
奇数:num[(n+1)/2]
偶数:(num[n/2] + num[n/2-1])/2
在这个程序中,利用了c语言整数相除向下取整的特性,将偶数的变了一下,变成了:
(num[(n+1)/2] + num[(n+1)/2-1])/2
因此index设置的是(nums1Size+nums2Size+1)/2
换一种设置也行,就是最后的处理也需要跟着变一下。

程序里的一个小小的细节,在设置左右边界的时候:

int left=0,right=smallsize;

题解

和二分查找那里不一样,这个程序的查找范围就是[0,smallsize],那条线是可以划到程序的头上的,所有有这个。

double max(double a,double b)
{
    if(a>b)return a;
    else return b;
}
double min(double a,double b)
{
    if(a<b)return a;
    else return b;
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){

    int index = (nums1Size+nums2Size+1)/2;
    int* smallone,*bigone,smallsize,bigsize;
    smallone = nums1Size<nums2Size?nums1:nums2;
    bigone =  nums1Size>=nums2Size?nums1:nums2;
    smallsize = nums1Size<nums2Size?nums1Size:nums2Size;
    bigsize = nums1Size>=nums2Size?nums1Size:nums2Size;

    int start = smallsize/2;
    int start_t = index-start;
    int left=0,right=smallsize;
    while(left<=right){
        start = left + (right - left)/2;
        start_t = index - start;
        printf("%d %d %d %d\n",left,right,start,start_t);
        if(start>0 && smallone[start-1]>bigone[start_t])
            right = start - 1;
        else if (start!=smallsize && smallone[start] < bigone[start_t-1])
            left = start + 1;
        else
            break;
    }
    double small,big;
    if(start == 0)
        small = bigone[start_t-1];
    else if (start_t == 0)
        small = smallone[start-1];
    else
        small = max(bigone[start_t-1],smallone[start-1]);

    if((nums1Size+nums2Size)%2==1)return small;
    if(start == smallsize)
        big = bigone[start_t];
    else if(start_t == bigsize)
        big = smallone[start];
    else
        big = min(smallone[start],bigone[start_t]);
    return (small+big)/2;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值