微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

leetcode 4 - binary search

分享图片

注意:

1)需要保证nums1 的长度比 nums2 的长度小;(否则vector指针会越界)

2)  当分割线(partition)在首或尾时,用INT_MIN 和 INT_MAX 代替。

思路:

分享图片

 

class Solution {
public:
    double static findMedianSortedArrays(vector<int>& nums1,vector<int>& nums2) {

        int x = nums1.size();
        int y = nums2.size();
        
        if(x>y)
            return findMedianSortedArrays(nums2,nums1);
        
        int l = x + y;
        int length = (x + y + 1) / 2;
        double median = 0;
        //vector x 中:
        int start = 0;
        int end = x;

        while (start <= end) {
            //cout << start << endl << end << endl;
            int p_x = (start + end) / 2;
            int p_y = length - p_x;

            //if p_x is 0 it means nothing is there on left side,use -INF for maxLeftX
            //if p_x is length of input then there is nothing on right side,use +INF for minRightX
            double maxLeftX = (p_x == 0) ? INT_MIN : nums1[p_x - 1];
            double minRightX = (p_x == x) ? INT_MAX : nums1[p_x];

            double maxLeftY = (p_y == 0) ? INT_MIN : nums2[p_y - 1];
            double minRightY = (p_y == y) ? INT_MAX : nums2[p_y];

            if (maxLeftX <= minRightY && maxLeftY <= minRightX)
            {
                if (l % 2 == 0)
                    //长度为偶数
                    {
                        median = (max(maxLeftX,maxLeftY)+ min(minRightX,minRightY)) / 2.0;
                        //cout << max(maxLeftX,maxLeftY) << endl << min(minRightX,minRightY) << endl;
                    }
                else
                    median = max(maxLeftX,maxLeftY);
                return median;
            }
            else if (maxLeftX > minRightY)
                end = p_x - 1;       //nums1的分割线左移
            else if (maxLeftY > minRightX)
                start = p_x + 1;   //nums1的分割线右移
        }
        return -1;
    }
};

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐