LeetCode:寻找两个正序数组的中位数

给定两个大小分别为 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
-106 <= nums1[i], nums2[i] <= 106
进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?

来源:力扣(LeetCode)
链接:(https://leetcode-cn.com/problems/median-of-two-sorted-arrays)

思路:

这道题目很容易想到先对两个数组进行排序,然后选取中位数,但是这样会额外需要O(m+n)空间,那有什么方法可以不用额外空间呢,思路其实和讲两个数组进行排序后再取中位数有异曲同工之处,我是否可以这样想,我们一刀是否能将两个数组切成两半,刚好使右边的数字都比左边的数字大且两边数量一致,那么我们这一刀上两边的数字就是我们需要的中间数字了,当两数组之和为偶数的时候,取左边最大的数、右边最小的数,再求平均就是我们的中位数了,当两数数组之和为奇数的时候,那就要看我们的切法是左边比右边多一个数字呢还是右边多一个数字,如果是左边多一个数字那么取左边最大的数字就是我们的中位数反之右边最小的数字就是我们的中位数了,请看下面的图解:

例1:两数组数目之和为偶数的时候

偶数时候的切割

解析:可以看得出来这一刀刚好将两个数组切为两个数量相等的部分,且右边的数字都比左边的大(只要是数组2切线左边最大的数字比数字1切线右边最小的数字小该条件或数组2切线右边最小的数字比数组1切线左边最大的数组大,都成立),我们取切线左边最大的数字8,再取切线右边最小的数字9,求平均(8+9)/2 = 8.5 刚好就是我们要的中位数。

例2:两数组数目之和为奇数的时候(这里我们的切法是使左边比右边多一个数)


奇数时候的切割

解析:可以看得出来这一刀刚好将两个数组切成左部分数目比右部分数目多一个,且右边的数字都比左边的大,我们这时候取切线左边最大的数字9,这个左部分最大的数字9刚好就是我们求的中位数。

代码:

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    vector<int> temp;
    int left,right,i,j;//left左边界 ,right边界 用于数组1的二分查找,i指向数组1元素的位置,j指向数组2元素的位置
    if (nums1.size()>nums2.size()) //这里为了让数组2的数量始终大于或等于数组1的数量(以防切割的时候数组2下没有元素,因为我们是遍历数组1来切割的) 
    { 
        temp = nums1;
        nums1 = nums2;
        nums2 = temp;
    }
    int m ,n, totalLeft;//totalLeft为切线数组1加上数组2左部分的数目
    m = nums1.size();
    n = nums2.size();
    left = 0;
    right = m;
    totalLeft = m + ((n - m) + 1) / 2; // (n+m+1)/2 的变式,这里加一是方便奇偶数统一运算
    while (left < right) 
    {
        i = left + (right - left) / 2;//二分查找 i为数组1查找的元素
        j = totalLeft - i; //那么j就一定要是切线左部分的数目减去数组1左部分的数目了
        if (nums2[j-1]>nums1[i])//如果数组2切线左边的数字大于数组1切线右边的数字,那么一定是数组1的数字不合适太小了
        {
            left = i+1;//左边界移至刚刚查找的地方,继续二分查找
        }
        else
        {
            right = i; //查找成功,将右边界至为i,退出二分查找
        }
    }
     i = left;//找到i的位置
     j = totalLeft - i;
    int nums1RightMin = i == m ? INT_MAX :nums1[i]; //这里需要考虑边界问题,如果找到的切线位于数组1最右端,那么使数组1右边最小值为最大值,这样就不会取到了
    int nums1LeftMax = i == 0 ? INT_MIN : nums1[i - 1];//如果找到的切线位于数组1最左端,那么使数组1左边最大值为最小值,这样就不会取到了
    int nums2RightMin = j == n ? INT_MAX : nums2[j];
    int nums2LeftMax = j == 0 ? INT_MIN : nums2[j - 1];
    if ((n+m)%2==1)
    {
        return (double)1.0*max(nums1LeftMax, nums2LeftMax) ; //如果两个数组数目之和为奇数,那么取数组1、数组2切线左边最大的值即可
    }
    else
    {
        return (double)1.0*(max(nums1LeftMax, nums2LeftMax)+ min(nums1RightMin,nums2RightMin))/2;//如果两个数组数目之和为偶数,那么取数组1、数组2切线左边最大的值与切线右边最小值得平均数
    }

}
int main() {
    vector<int> nums1, nums2;

    nums1.push_back(2);
    nums1.push_back(3);

    nums2.push_back(1);
    
    cout<<findMedianSortedArrays(nums1, nums2)<<endl;
    return 0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容