天题系列:Median of Two Sorted Array

答案部分完全借鉴 http://www.cnblogs.com/springfor/p/3861890.html

以下是摘抄:“爱做饭的小莹子

题目:

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

题解:

首先我们先明确什么是median,即中位数。

引用Wikipedia对中位数的定义:

计算有限个数的数据的中位数的方法是:把所有的同类数据按照大小的顺序排列。如果数据的个数是奇数,则中间那个数据就是这群数据的中位数;如果数据的个数是偶数,则中间那2个数据的算术平均值就是这群数据的中位数。

因此,在计算中位数Median时候,需要根据奇偶分类讨论。

解决此题的方法可以依照:寻找一个unioned sorted array中的第k大(从1开始数)的数。因而等价于寻找并判断两个sorted array中第k/2(从1开始数)大的数。

特殊化到求median,那么对于奇数来说,就是求第(m+n)/2+1(从1开始数)大的数。

而对于偶数来说,就是求第(m+n)/2大(从1开始数)和第(m+n)/2+1大(从1开始数)的数的算术平均值。

那么如何判断两个有序数组A,B中第k大的数呢?

我们需要判断A[k/2-1]和B[k/2-1]的大小。

如果A[k/2-1]==B[k/2-1],那么这个数就是两个数组中第k大的数。

如果A[k/2-1]<B[k/2-1], 那么说明A[0]到A[k/2-1]都不可能是第k大的数,所以需要舍弃这一半,继续从A[k/2]到A[A.length-1]继续找。当然,因为这里舍弃了A[0]到A[k/2-1]这k/2个数,那么第k大也就变成了,第k-k/2个大的数了。

如果 A[k/2-1]>B[k/2-1],就做之前对称的操作就好。

这样整个问题就迎刃而解了。

当然,边界条件页不能少,需要判断是否有一个数组长度为0,以及k==1时候的情况。

因为除法是向下取整,并且页为了方便起见,对每个数组的分半操作采取:

int partA = Math.min(k/2,m);
int partB = k - partA;

为了能保证上面的分半操作正确,需要保证A数组的长度小于B数组的长度。

同时,在返回结果时候,注意精度问题,返回double型的就好。

        //直接没什么想法,看了答案也觉得万万想不到 ref http://www.cnblogs.com/springfor/p/3861890.html, http://blog.csdn.net/linhuanmars/article/details/19905515
        // 找posA 和posB的部分略魔性
     public class Solution {
    public double findMedianSortedArrays(int A[], int B[]) {
        int m = A.length, n = B.length;
        int t=m+n, k = t/2;
        if(t%2!=0){
            return (double) helper(A,B,0,m-1,0,n-1,k+1);
        }else{
            int x = helper(A,B,0,m-1,0,n-1,k+1);
            int y = helper(A,B,0,m-1,0,n-1,k);
            return (double) (x+y)/2;
        }
    }
    public int helper(int A[], int B[], int i, int i2, int j, int j2, int k){
        int m = i2-i+1, n = j2-j+1;
        if(m==0) return B[k+j-1];
        if(m>n) return helper(B,A,j,j2,i,i2,k);
        if(k==1) return Math.min(A[i], B[j]);

        int posA = Math.min(m, k/2);
        int posB = k-posA;
        if(A[posA+i-1]==B[j+posB-1]){
            return A[posA+i-1];
        }else if(A[posA+i-1]<B[j+posB-1]){
            return  helper(A,B,i+posA,i2,j,j2,k-posA);
        }else{
            return  helper(A,B,i,i2,j+posB,j2,k-posB);
        }
    }
}
时间: 2024-10-17 02:51:50

天题系列:Median of Two Sorted Array的相关文章

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)). 这道题的

Median of Two Sorted Array leetcode java

题目: 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)). 题解: 首先我们先明确什么是median,即中位数. 引用Wikipedia对中位数的定义: 计算有限个数的数据的中位数的方法是:把所有的同类数据按照大小的顺序排列

【Leetcode】Median of Two Sorted Array II

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)). 非常经典的一个问题,如果直接查找中位数,即偶数约定向下取整,奇数就是严格的中位数,这个题目直接可以比较每个数组的mid来计算得到. 但是这个问题的需求更改为如果为偶数,则中

LeetCode 新题: Find Minimum in Rotated Sorted Array II 解题报告-二分法模板解法

Find Minimum in Rotated Sorted Array II Follow up for "Find Minimum in Rotated Sorted Array":What if duplicates are allowed? Would this affect the run-time complexity? How and why?Suppose a sorted array is rotated at some pivot unknown to you be

刷题4. Median of Two Sorted Arrays

一.题目 Median of Two Sorted Arrays,具体请自行搜索. 这个题目,我看了一下,经过一番思考,我觉得实现起来不是很复杂. 但要做到bug free也不难,最大的问题是性能问题. 性能只有42%的样子,内存占用太多.还需要进一步优化!!! 二.这个题目,我自己实现 提交了2次: 第1次: Wrong Answer 第2次:终于对了 下面是我的完整代码实现,需要的拿去: #include<iostream> #include<vector> using nam

LeetCode 4 Median of Two Sorted Array

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)). 这个题是求两个已排数组的中位数,更一般的就是求两个已排数组的第k大的数. 自己想到的解决方法有一个先用合并排序将两个数组进行排序得到一个排列后的数组,在遍历找到第k到的数

leetcode——Median of Two Sorted Array

原题为: 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)). 好吧,我是普通青年,第一反应就是使用Merge两个数字的思想,直接找到中位数.但是这个算法的复杂度是O(m+n). leetcode给这道题打上的

算法题之Median of Two Sorted Arrays

这道题是LeetCode上的题目,难度级别为5,刚开始做没有找到好的思路,以为是自己智商比较低,后来发现确实也比较低... 题目: 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)). Example 1: n

leetcode4 median Of two sorted array

leetcode4 此题归在二分法中: 大概思想:将两个数组分别分为两个部分,两者的数值较小的数字,就是合并后数组的前半部分 难点在:找到第一个数组的位置i,则第二数组的位置j = (m+n +1)/2 - i; 此后,由i,j找到中位数,如果m+n为奇数,则max_left即为所求,否则:(max_left + min_right)/2为结果 class Solution { public: double findMedianSortedArrays(vector<int>& num