Leetcode题解4.Median of Two Sorted Arrays

Leetcode题解

4.Median of Two Sorted Arrays-Hard

Author: Zinc
关键词:分治,递归,二分

题目

There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

中文简述

有两个有序数组nums1和nums2,长度分别为n和m,找出两数组合并在一起的中位数。特别要求:时间复杂度为O(log(m+n))

分析

线性时间:
看到题目有人就会开始说,这道题很简单嘛,直接将两个有序数组合并成一个有序数组然后用中位数的定义:中位数为有序数组中间那个数(长度为奇数)或者是中间两个数的平均数(长度为偶数时)。由于原来给定的两个数组已经是有序数组,所以我们只要如同归并排序的操作一步即可,时间复杂度为m+n,线性的遍历一遍排好序就可以了。
最后用公式整个的步骤是m+n+1步,时间复杂度为O(m+n),线性的时间貌似已经很优秀了,题目画下一个完美的句号。
“喂喂喂,你瞎么。。。题目让你用log(m+n)的时间哇!“(答题的时候从旁边传来不屑的嘲讽)
“什么!喵?喵?喵?”(此时的我一脸黑人问号),定睛一看真的是log(m+n),上天绕过谁。果然这道Hard的题目不是好惹的。

对数时间分析:
看到对数时间相信很多人和我一样,一开始是懵逼的,怎么做?其实,就对数时间这一点而言,其实能达到这么快速的算法不多,最常见的就是二分的思想。好就以二分为切入点往下扩展,所谓的找到中位数,其实就是找到第k=len/2(向上取整)的数或者k=len/2与k=len/2+1的两个数的平均数,问题转变为找2个有序数组中第K大的数字,然后通过二分不段减小两个数组的规模。就得到了我们求解问题的代码。
简单的流程是:
1。从左右两边找到第K大的数(初始的K就是中位数的位置)
2。分别找两个数组k/2位置的数字进行比较,
若a1[k/2]>a2[k/2]说明,a1前k/2在搜素范围外,反之则a2的前k/2。
(k/2的长度和a1长度取最小那个进行算法;当a1无数字则答案在b中,若相等则答案就为相等的数字。)
3。情况1就将a1去除前k/2,搜索两个数组中第k/2大的数字
情况2就像a2去除前k/2,搜索两个数组中第k/2大的数字

Source Code

1.归并

    class Solution {
    public:
        double med_singlevec(vector<int>& nums){
            int len = nums.size();
            int mid = len/2;
            return ((len%2==0)? ((nums[mid]+nums[mid-1])/2.0) : nums[mid]);
        }

        double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
            int i = 0, j = 0;
            vector<int> nums;
            for(i = 0, j = 0; i<nums1.size()&&j<nums2.size();){
                if(nums1[i]<=nums2[j]){
                    nums.push_back(nums1[i++]);
                }
                else
                    nums.push_back(nums2[j++]);
            }
            while(i<nums1.size())
                nums.push_back(nums1[i++]);
            while(j<nums2.size())
                nums.push_back(nums2[j++]);
            return  med_singlevec(nums) ;


        }

2.二分分治

    #include <cmath>
    class Solution {
    public:
        double findKthnum(vector<int>&a, vector<int>&b, int k){
            int alen = a.size(), blen = b.size();
            if(alen > blen)
                return findKthnum(b, a, k);
            if(alen==0)
                return b[k-1];
            if(k==1)
                return min(a[0],b[0]);
            int pa = min(k/2, alen);
            int pb = k - pa;
            if(a[pa-1] < b[pb-1]){
                vector<int>*na = new vector<int>(a.begin()+pa, a.begin()+alen);
                return findKthnum(*na,b,k-pa);
            }

            else if(a[pa-1] > b[pb-1]){
                vector<int>* nb = new vector<int>(b.begin()+pb,b.begin()+blen);
                return findKthnum(a,*nb,k-pb);
            }

            else
                return a[pa-1];

        }
        double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
            int total = nums1.size()+nums2.size();
            if(total % 2 == 1){
                return findKthnum(nums1, nums2, total/2 +1);
            }
            else{
                return (findKthnum(nums1, nums2, total/2)+
                        findKthnum(nums1, nums2, total/2+1))/2.0;
            }
        }
    };
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值