首页 > 其他 > 详细

leetcode第一刷_Median of Two Sorted Arrays

时间:2014-05-15 12:21:25      阅读:439      评论:0      收藏:0      [点我收藏+]

这道题是我最初刷的那20多道之一,但一直没有过,被各种各样的情况折磨死了,直到把所有其他的题都写完,回来看大神对这道题是怎么处理的时候,才惊叹算法的奇妙。再次验证了我的想法,如果要处理各种各样的特殊情况,一定是算法本身有问题。

之前看过很多有关在两个排序数组中找中位数的解法,大多根据两个数组长度不同分了很多种情况,各种讨论。下面要介绍的方法并没有直接求中位数,而是把求中位数转换成了求两个数组合并之后的第(m+n)/2个数(当然根据总数的奇偶有所不同,不过思路上是没有问题的)。用k来表示进入递归的时候,应该求的两个数列合并之后的那个数,用二分的思想,我们先看A[k/2]与B[k/2]的大小关系,如果A[k/2-1]<B[k/2-1],那么A[0,k/2-1]中有没有可能存在合并之后的第K个数呢?A[k/2-1]大于A中在它之前的k/2-1个数,最好的情况,它也大于B中的在B[k/2-1]之前的k/2-1个数,也就是A[k/2-1]最大在合并后的数列中可以排到第(k/2-1+k/2-1+1)个位置,也即第k-1个位置。因此A[k/2-1]这一部分肯定不包含结果,应该直接扔掉。

对称的,当A[k/2-1]>B[k/2-1]时,有类似的结果。

当A[k/2-1] == B[k/2-1]时呢?说明这个数找到了,返回即可。

在实现的时候有几个技巧,两个数列个数不一样,为了下面编码部分的简便,可以在开头判断一下,如果前面数组个数多,就互换一下。第二点,数组中剩余的的元素可能比k/2要少,这时候就取一个较小的值即可。至于最开始两个数组合并后元素个数奇偶的问题,正是由于我们求的是第k个数而不是中位数,大大简化 了。设总数为t,如果是奇数,就求第t/2+1的那个数,如果是偶数,就分别求第t/2和第t/2+1的数,然后求平均。

28行代码完成了我100多行没有完成的事情,算法太美。

class Solution {
public:
    double findKth(int *A, int m, int *B, int n, int k){
        if(m>n){
            return findKth(B, n, A, m, k);
        }
        if(m == 0)
            return B[k-1];
        if(k == 1)
            return min(A[0], B[0]);
        int pa = min(m, k/2), pb = k-pa;
        if(A[pa-1]<B[pb-1]){
            return findKth(A+pa, m-pa, B, n, k-pa);
        }else if(A[pa-1]>B[pb-1]){
            return findKth(A, m, B+pb, n-pb, k-pb);
        }else{
            return A[pa-1];
        }
    }
    double findMedianSortedArrays(int A[], int m, int B[], int n) {
        int t = m+n;
        if(t&0x1){
            return findKth(A, m, B, n, t/2+1);
        }else{
            return (findKth(A, m, B, n, t/2)+findKth(A, m, B, n, t/2+1))/2;
        }
    }
};


leetcode第一刷_Median of Two Sorted Arrays,布布扣,bubuko.com

leetcode第一刷_Median of Two Sorted Arrays

原文:http://blog.csdn.net/u012792219/article/details/25871077

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!