4.Median of Two Sorted Arrays(Array; Divide-and-Conquer)

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(logx)的算法,就想到二分法。二分法结束的条件是任何一个array只剩下一个元素了。每次递归(二分),去除某个array的一半。另外注意,每次二分至少要去掉一个元素。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int size1 = nums1.size();
        int size2 = nums2.size();
        int size = size1 + size2;
        int m = size >> 1;

        if(nums1.empty()){
            if(size%2 == 1) return nums2[m];
            else return (double) (nums2[m-1] + nums2[m])/2;
        }
        if(nums2.empty()){
            if(size%2 == 1) return nums1[m];
            else return (double) (nums1[m-1] + nums1[m])/2;
        } 

        if(size%2 == 1) return findK(nums1, nums2, 0, size1-1, 0, size2-1, m);
        else return (double)(findK(nums1, nums2, 0, size1-1, 0, size2-1, m-1)+findK(nums1, nums2, 0, size1-1, 0, size2-1, m))/2;
    }

    int findK(vector<int>& nums1, vector<int>& nums2,int s1, int e1, int s2, int e2, int k){
        if(s1 == e1){
            if(k == 0) return min(nums1[s1],nums2[s2]); //为了防止s2+k-1越界
            if(nums1[s1] < nums2[s2+k-1]) return nums2[s2+k-1];
            else if(s2+k-1 == e2) return nums1[s1]; //为了防止s2+k越界
            else return min(nums1[s1], nums2[s2+k]);
        }
        if(s2 == e2){
            if(k == 0) return min(nums1[s1],nums2[s2]); //为了防止s1+k-1越界
            if(nums2[s2] < nums1[s1+k]) return max(nums2[s2],nums1[s1+k-1]);
            else if(s1+k-1 == e1) return nums2[s2]; //为了防止s1+k越界
            else return min(nums2[s2], nums1[s1+k]);
        }

        int m1 = (s1+e1)>>1;
        int m2 = (s2+e2)>>1;

        int halfLen = (e1 - s1 + e2 - s2 + 2) >> 1;
        if(k >= halfLen){ //k is in the second half
            if(nums1[m1] < nums2[m2]){ //delete first half of num1
                return findK(nums1, nums2, m1+1, e1, s2, e2, k-(m1-s1+1)); //+1是考虑到m1==s1的情况,每次二分至少要去除一个元素
            }
            else{ //delete fist half of num2
                return findK(nums1, nums2, s1, e1, m2+1, e2, k-(m2-s2+1));//+1是考虑到m2==s2的情况,每次二分至少要去除一个元素
            }
        }
        else{ //k is in the first half
            if(nums1[m1] < nums2[m2]){ //delete second half of num2
                return findK(nums1, nums2, s1, e1, s2, m2, k);
            }
            else{ //delete second half of num1
                return findK(nums1, nums2, s1, m1, s2, e2, k);
            }
        }
    }
};
时间: 2024-10-27 11:13:10

4.Median of Two Sorted Arrays(Array; Divide-and-Conquer)的相关文章

LeetCode 第二题,Median of Two Sorted Arrays

题目再现 There are two sorted arrays A and B 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)). 题意解析 题目意思是给两个大小为m,n的有序数组(m,n可能为0),要求找出这两个数组的中位数.并且程序的时间复杂度必须不能超过O(log(m+n)). 这道题的

[LintCode] Median of Two Sorted Arrays 两个有序数组的中位数

There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. Have you met this question in a real interview? Example Given A=[1,2,3,4,5,6] and B=[2,3,4,5], the median is 3.5. Given A=[1,2,3] and B=[4,5],

LeetCode(3) || Median of Two Sorted Arrays

LeetCode(3) || Median of Two Sorted Arrays 题记 之前做了3题,感觉难度一般,没想到突然来了这道比较难的,星期六花了一天的时间才做完,可见以前基础太差了. 题目内容 There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should

LeetCode——Median of Two Sorted Arrays

There are two sorted arrays A and B 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)). 有两个已排序的数组A和B,大小为m 和 n. 找出两数组的中位数 时间复杂度应该为 O(log (m+n)). 简单粗暴的方法就是把两个数组合并成一个数组,排序,取中位数.

第三周 leetcode 4. Median of Two Sorted Arrays (HARD)

4. Median of Two Sorted Arrays 给定两个有序的整数序列.求中位数,要求复杂度为对数级别. 通常的思路,我们二分搜索中位数,对某个序列里的某个数 我们可以在对数时间内通过二分算法求得两个序列中比它小的数,整体复杂度也是对数级别.但是代码实现较为困难. 换一个思路,我们把中位数不要当作一个数,而当作一个序列的划分.划分后,序列的左半部设为L,右半部设为R 满足max(L)<=min(R)且满足len(L)==len(R) 二分搜索这个划分即可.对于A+B的长度为奇数的情

leetcode 2.Median of Two Sorted Arrays

Median of Two Sorted Arrays Problem: There are two sorted arrays A and B 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)). 解题思路:如果是一个已排序数组,求中位数,可以直接计算.因此比较常规的想法是直接将两个有序数组合并

【题解】【数组】【Leetcode】Median of Two Sorted Arrays

Median of Two Sorted Arrays There are two sorted arrays A and B 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)). ? 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

周刷题第二期总结(Longest Substring Without Repeating Characters and Median of Two Sorted Arrays)

这周前面刷题倒是蛮开心,后面出了很多别的事情和问题就去忙其他的,结果又只完成了最低目标. Lonest Substring Without Repeating Characters: Given a string, find the length of the longest substring without repeating characters. Examples: Given "abcabcbb", the answer is "abc", which t

[leetcode]Median of Two Sorted Arrays @ Python

原题地址:https://oj.leetcode.com/problems/median-of-two-sorted-arrays/ 题意:There are two sorted arrays A and B 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)). 解题思路:这道题要求两个已经排好