代码随想录算法训练营|二分法相关题目

LeetCode 35 搜索插入位置

题目链接：https://leetcode.cn/problems/search-insert-position/

题目描述：给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。nums为无重复元素的升序排列数组

示例1：
输入: nums = [1,3,5,6], target = 5
输出: 2

示例2：
输入: nums = [1,3,5,6],
target = 2
输出: 1

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

左闭右闭

class Solution {//左闭右闭
    public int searchInsert(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        while(left <= right){
            int middle = left + (right - left)/2;
            if(nums[middle] > target){
                right = middle -1;
            }else if(nums[middle] < target){
                left = middle + 1;
            }else{
                return middle;
            }
        }
        return right + 1;//left==right还未找到则插入right+1
    }
}

左闭右开

class Solution {//左闭右开
    public int searchInsert(int[] nums, int target) {
        int left = 0, right = nums.length;
        while(left < right){
            int middle = left + (right - left)/2;
            if(nums[middle] > target){
                right = middle;
            }else if(nums[middle] < target){
                left = middle + 1;
            }else{
                return middle;
            }
        }
        return right;
    }
}

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

题目链接：https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/

题目描述：给你一个按照非递减顺序排列的整数数组 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]

示例3：
输入：nums = [], target = 0
输出：[-1,-1]

思路：寻找target在数组里的左右边界，有如下三种情况：

• 情况一：target 在数组范围的右边或者左边，例如数组{3, 4, 5}，target为2或者数组{3, 4, 5},target为6，此时应该返回{-1, -1}
• 情况二：target 在数组范围中，且数组中不存在target，例如数组{3,6,7},target为5，此时应该返回{-1, -1}
• 情况三：target 在数组范围中，且数组中存在target，例如数组{3,6,7},target为6，此时应该返回{1, 1}

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int leftBorder = getLeftBorder(nums, target);
        int rightBorder = getRightBorder(nums, target);
        if(leftBorder ==-2 || rightBorder == -2) return new int[]{-1,-1};
        if(rightBorder - leftBorder > 1){
            return new int[]{leftBorder + 1, rightBorder - 1};
        }
        return new int[]{-1,-1};
    }

    int getRightBorder(int[] nums,int target){
        int left = 0, right = nums.length - 1;
        int rightBorder = -2;
        while(left <= right){
            int middle = left + (right - left)/2;
            if(nums[middle] > target){
                right = middle - 1;
            }else{// 寻找右边界，nums[middle] == target的时候更新left
                left = middle + 1;
                rightBorder = left;
            }
        }
        return rightBorder;
    }

    int getLeftBorder(int[] nums, int target){
        int left =0, right = nums.length - 1;
        int leftBorder = -2;
        while(left <= right){
            int middle = left + (right - left)/2;
            if(nums[middle] >= target){
                right = middle - 1;
                leftBorder = right;
            }else{
                left = middle + 1;
            }
        }
        return leftBorder;
    }
}

LeetCode 367 有效的完全平方数

题目链接：https://leetcode.cn/problems/valid-perfect-square/

题目描述：给你一个正整数 num 。如果 num 是一个完全平方数，则返回 true ，否则返回 false 。 完全平方数
是一个可以写成某个整数的平方的整数。换句话说，它可以写成某个整数和自身的乘积。 不能使用任何内置的库函数，如 sqrt 。

示例1：
输入：num = 16
输出：true
解释：返回 true ，因为 4 * 4 = 16 且 4 是一个整数。

示例2：
输入：num = 14
输出：false
解释：返回 false ，因为 3.742 * 3.742 = 14 但 3.742不是一个整数。

class Solution {
    public boolean isPerfectSquare(int num) {
        int left = 0, right = num;
        while(left <= right){
            int middle = left + (right - left)/2;
            long sqr = (long)middle * middle;
            if(sqr > num){
                right = middle - 1;
            }else if(sqr < num){
                left = middle + 1;
            }else{
                return true;
            }
        }
        return false;
    }
}

LeetCode 69 x的平方根

题目链接：https://leetcode.cn/problems/sqrtx/

题目描述：给你一个非负整数 x ，计算并返回 x 的算术平方根 。由于返回类型是整数，结果只保留 整数部分 ，小数部分将被舍去 。注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例1：输入：x = 4 输出：2

示例2：输入：x = 8 输出：2
解释：8 的算术平方根是 2.82842…, 由于返回类型是整数，小数部分将被舍去。

class Solution {
    public int mySqrt(int x) {
        int left = 0, right = x, ans = -1;
        while(left <= right){
            int middle = left + (right - left)/2;
            if(((long)middle * middle) <= x){
                ans = middle;
                left = middle + 1;
            }else{
                right = middle - 1;
            }
        }
        return ans;
    }
}