Leetcode(4) Median of Two Sorted Arrays

本文章主要是对于leetcode网站的算法题目4进行总结。原题如下:

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)).

主要是针对两个已经排列好的整数数组,找出其中的中位数,并且要求时间复杂度为O(log (m+n)).

1.首先是中位数的定义。中位数是将一个数的集合分成两个相等的子集,并且一个子集里的任何一个数总是大于另一个子集的任何一个数。

2.对于两个整数数组(A,B),如果我们将其放到一个集合里面,并且利用两个分割线(1个在A,一个在B)将集合分成相等的两部分,如下图
这里写图片描述
我们只需要保证两个条件:
****1. len(left_part)== len(right_part)
2. max(left_part)<= min(right_part)**
那么我们就能将集合A,B里的元素分成相等的两部分,而且left_part的任何一个数总是小于right_part的任何一个数。那么中位数=(max(left_part) + min(right_part))/2.

为了满足上面的条件,我们可以得到
(1)i+j=m-i+n-j,如果n>=m,我们只需要:i=0~m,j=(m+n+1)/2-i
(2)B[j-1]<=A[i]&&A[i-1]<=B[j]

那么问题就转换成我们只需找到一个‘i’(0<=i<=m),使得这个'i'满足条件(2)就可以了。
具体算法步骤如下:
这里写图片描述

C++代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size(), n = nums2.size();
        if (m > n) return findMedianSortedArrays(nums2, nums1);
        int i, j, imin = 0, imax = m, half = (m + n + 1) / 2;
        while (imin <= imax) {
            i = (imin & imax) + ((imin ^ imax) >> 1);
            j = half - i;
            if (i > 0 && j < n && nums1[i - 1] > nums2[j]) imax = i - 1;
            else if (j > 0 && i < m && nums2[j - 1] > nums1[i]) imin = i + 1;
            else break;
        }
        int num1;
        if (!i) num1 = nums2[j - 1];
        else if (!j) num1 = nums1[i - 1]; 
        else num1 = max(nums1[i - 1], nums2[j - 1]);
        if ((m + n) & 1) return num1;
        int num2;
        if (i == m) num2 = nums2[j];
        else if (j == n) num2 = nums1[i];
        else num2 = min(nums1[i], nums2[j]);
        return (num1 + num2) / 2.0;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值