二分查找练习

二分查找练习

leetcode35 - 查找插入位置

题目描述

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:
输入: [1,3,5,6], 5
输出: 2

示例 2:
输入: [1,3,5,6], 2
输出: 1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position

思路

这道题目要求在一个有序数组(nums)中查找一个目标值(target)或返回它被按顺序插入数组的位置，同时题目强调数组中无重复元素，则返回的值一定是唯一的。
在这里我们采用左闭右闭区间的二分法,即目标值在我们给定的区间 [left,right] 中，所以left和right的初值应对应nums数组里面第一个元素和最后一个元素的下标。
这里还应特别注意以下两点：

• 在 while(left<=right) 中，判断条件应是 <= 而不是 < ，因为当 left==right 是该区间也是有元素存在的。
• 若 nums[mid]>target，此时应修改 right 的值，因为已经判断过下标为 mid 时不符合条件，所以此时应把 right 的值修改为 mid-1 以提高程序的效率。

代码

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int mid,left=0;  //left赋值为nums数组第一个元素的下标
        int right=nums.size()-1; //right赋值为nums数组最后一个元素的下标
        while(left<=right){
            mid=left+(right-left)/2; //实际上与(left+right)/2没有本质区别,但可以有效防止越界
            if(nums[mid]<target){
                left=mid+1;
            }
            else if(nums[mid]>target){
                right=mid-1;
            }
            else{
                return mid; //找到与target值相同的元素，直接返回下标
            }
        }
        //没有找到与target值相同的元素，但此时mid对应的元素的值与target最接近
        if(nums[mid]<target){ 
            return mid+1;
        }
        else{
            return mid;
        }
    }

};

leetcode34 -在排序数组中查找第一个和最后一个元素的位置

题目描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

进阶：

你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

示例 1：
输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：
输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array

思路

思路与上题基本一致，在数组中找到与目标元素相同的值的位置后(任意一个都可以)，可以让 left=mid，right=mid，然后令left左移找到起始位置，令right右移找到结束位置。

代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int mid,left=0;
        int right=nums.size()-1;
        while(left<=right){
            mid=left+(right-left)/2;
            if(target>nums[mid]){
                left=mid+1;
            }
            else if(target<nums[mid]){
                right=mid-1;
            }
            else{
                break; //找到元素位置后还需要再做处理，所以直接break
            }
        }
        vector<int> vi;
        if(left>right){  //left>right说明没有找到目标元素
            vi.push_back(-1);
            vi.push_back(-1);
            return vi;
        }
        // 程序能执行到这里说明目标元素已经找到，接下来处理一下即可
        left=mid;
        right=mid;
        while(left-1>=0&&nums[left-1]==target){ //注意left-1>=0应放在前面先判断，防止越界
            left--;
        }
        while(right+1<nums.size()&&nums[right+1]==target){ //与上面同理
            right++;
        }
        vi.push_back(left);
        vi.push_back(right);
        return vi;
    }
};