Leetcode 34 - search for a Range（二分）

最新推荐文章于 2018-07-05 10:04:47 发布

原创最新推荐文章于 2018-07-05 10:04:47 发布 · 287 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode

leetcode 同时被 2 个专栏收录

157 篇文章

订阅专栏

二分

27 篇文章

订阅专栏

本文介绍了一个使用二分查找算法解决的问题：在一个有序数组中找到特定目标值首次和末次出现的位置。通过两个定制化的二分查找函数，分别定位目标值的左边界和右边界。

题意

给定一个有序数组a[]，和一个目标数字target，求target在a[]中出现的范围。

思路

最先想用lower_bound和upper_bound随便搞一样，但是想了一样既然在学二分就好好写二分吧。其实这个题里面两个二分的区别就在于取等号的不同。

细节

当a[m] = target的时候不能直接返回，因为我们要找的是最边界的那两个点。

代码

class Solution {
public:
    int l_bs(vector<int>& a, int target) {
        int l = 0, r = a.size() - 1, m, ans = -1;
        while (l <= r) {
            m = l + (r - l >> 1);
            if (a[m] < target) l = m + 1;
            else r = m - 1, ans = m;
        }
        return ans >= 0 ? (a[ans] == target ? ans : -1) : -1;
    }

    int h_bs(vector<int>& a, int target) {
        int l = 0, r = a.size() - 1, m, ans = -1;
        while (l <= r) {
            m = l + (r - l >> 1);
            if (a[m] <= target) l = m + 1, ans = m;
            else r = m - 1;
        }
        return ans >= 0 ? (a[ans] == target ? ans : -1) : -1;
    }

    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;
        ans.push_back(l_bs(nums, target));
        ans.push_back(h_bs(nums, target));
        return ans;
    }
};