• Leetcode 4 Median of Two Sorted Arrays 二分查找(二分答案+二分下标)


    貌似是去年阿里巴巴c++的笔试题,没有什么创新直接照搬的。。。

    题意就是找出两个排序数组的中间数,其实就是找出两个排序数组的第k个数。

    二分答案,先二分出一个数,再用二分算出这个数在两个排序数组排序第几,然后和k做比较,最后以这个比较为依据接着二分直到求出第k个数。

    本来这是两重二分,由于楼主的懒已经没有办法治疗了,用库函数upper_bound()代替了第二重二分,有志者可以自己写第二重二分,楼主就偷懒了。。。

    警告:由于题目很神奇,会出现两个排序数组为空的情况,楼主差点因为这个报警。。。。

     1 class Solution {
     2 public:
     3     int findkth(vector<int>& nums1, vector<int>& nums2, int k){
     4         int l = min(nums1[0],nums2[0]);
     5         int r = max(nums1[nums1.size() - 1], nums2[nums2.size() - 1]);
     6         while(l <= r){
     7             int mid = l + (r - l)/ 2;
     8             int k1 = upper_bound(nums1.begin(), nums1.end(), mid) -  nums1.begin() ;
     9             int k2 = upper_bound(nums2.begin(), nums2.end(), mid) -  nums2.begin() ;
    10             if(k1 + k2 == k){
    11                 if(k1 > 0 && mid == nums1[k1 - 1]) return mid;
    12                 else if(k2 > 0 && mid == nums1[k2 - 1]) return mid;
    13                 else r = mid - 1;
    14             }
    15             else if(k1 + k2 > k){
    16                 r = mid - 1;
    17             }
    18             else l = mid + 1;
    19         }
    20         return l;
    21     }
    22     double getmid(vector<int>& nums){
    23         if(nums.size() % 2 == 0) return (nums[nums.size() / 2 - 1] + nums[nums.size() / 2]) / 2.0;
    24         else return nums[nums.size() / 2];
    25     }
    26     double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    27         if(nums1.size() == 0 && nums2.size() == 0) return 0.0;
    28         if(nums1.size() == 0) return getmid(nums2);
    29         if(nums2.size() == 0) return getmid(nums1);
    30         int k = (nums1.size() + nums2.size() + 1) / 2;
    31         if((nums1.size() + nums2.size()) % 2 == 0){
    32             return (findkth(nums1, nums2, k) + findkth(nums1, nums2, k + 1))/2.0; 
    33         } 
    34         else return findkth(nums1, nums2, k);
    35     }
    36 };
  • 相关阅读:
    boost pool 和 object_pool
    boost::ref
    Source Insight设置
    windows7下硬盘安装ubuntu14.04
    POJ 2778 DNA Sequence (AC自己主动机 + dp)
    cocos2dx 运动+旋转动画 CCSequence CCAnimation CCAnimate CCMoveTo CCCallFuncN
    LeetCode 2 Add Two Numbers
    MySQL数据库导入外部*.sql文件具体步骤
    [ExtJS5学习笔记]第十节 Extjs5新增特性之ViewModel和DataBinding
    svn项目导入
  • 原文地址:https://www.cnblogs.com/onlyac/p/5584666.html
Copyright © 2020-2023  润新知