4.Median of Two Sorted Arrays

这个题好难,主要是有时间复杂度 O(log(m+n)) 的限制。

思路:
把问题转化成求两个数组中求第K小的数即可。
用分治法,若A[k/2-1] >B[k/2-1] (即A的第k/2个数大于B的第k/2个数),则第K小的数不在B的前k/2个数中,同理B>A。
由此,每进行比较一次,都会有k/2个数被筛选下去,一直到最后就可以找到了。

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int k = nums1.size() + nums2.size();
        if(k%2==0) return ( findKth(nums1,0,nums2,0,k/2) + findKth(nums1,0,nums2,0,k/2+1)) / 2.0;
        else return findKth(nums1,0,nums2,0,k/2+1);
    }
    
    int findKth(vector<int>& nums1,int i,vector<int>& nums2,int j,int k){
        if(i >= nums1.size()) return nums2[j + k-1];
        if(j >= nums2.size()) return nums1[i + k-1];
        if(k == 1) return min(nums1[i],nums2[j]);
        int a = i+k/2-1 >= nums1.size() ? INT_MAX :nums1[i+k/2-1]; 
        int b = j+k/2-1 >= nums2.size() ? INT_MAX:nums2[j+k/2-1];
        if(a >= b) return findKth(nums1,i,nums2,j+k/2,k-k/2);
        else return findKth(nums1,i+k/2,nums2,j,k-k/2);
    }

注:开始智障了,没有用INT_MAX,用的是10000,结果没通过,还找了很久的bug.....


运行结果

太慢了,还可以进行优化的(数组的K/2位置的数字要不要去除)。等研究完最快的代码再来填坑。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 一切都是那么的陌生,自己一个人来到一个陌生的地方,以为自己一切可以重新开始。那只是以为而已,当我来到这个陌生的地方...
    蔬书菜阅读 346评论 2 0
  • 宁夏脚踢、拍打、扔拖鞋,老公还是一脸坏笑地嘲弄她:“想要手机,自己来拿啊,哈哈哈!有本事把门踢破。” 老公趁宁夏一...
    宁静的夏天2017阅读 212评论 0 0
  • 龟速移动的车流,无法忍受的习惯着。一幢幢高楼,闪过不能聚焦的瞳孔,突然,定在一颗笔直的树上。在灰黄裸露的工...
    不恌阅读 625评论 0 2