2021-04-26

LeetCode:704. 二分查找

题目链接：https://leetcode-cn.com/problems/binary-search/

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target  ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

 

提示：

1. 你可以假设 nums 中的所有元素是不重复的。
2. n 将在 [1, 10000]之间。
3. nums 的每个元素都将在 [-9999, 9999]之间。

二分查找的算法比较好理解，但是写的时候并不好写，会有很多的边界问题，所以最好是背一下二分查找的模板。我分别用两个模板完成这道题目，这道题所提供的元素都不重复。如果元素有重复，那么第一个模板返回的是该元素第一个的下标，第二个模板返回的是该元素最后一个的下标。

用第一个模板完成的代码：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        int mid;
        while(l<r)
        {
            mid=(l+r)>>1;
            if(nums[mid]>=target) r=mid;
            else l=mid+1;
        }
        if(nums[l]==target) return l;
        else return -1;
    }
};

用第二个模板完成的代码：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        int mid;
        while(l<r)
        {
            mid=(l+r+1)>>1;
            if(nums[mid]<=target) l=mid;
            else r=mid-1;
        }
        if(nums[l]==target) return l;
        else return -1;
    }
};