数组之二分法

刷题小记

力扣35题-搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。

数组之二分法_第1张图片
数组之二分法_第2张图片
这题是在二分法基础上完成的!
第一种情况:刚好找到,返回下标即可!
第二种情况:不存在,
1,target比nums[mid]要小,此时插入mid元素之前,mid后移一位,所以新元素下标为mid。
2,target比nums[mid]要大,此时插入mid元素之后,mid不动,所以新元素下标为mid+1。

代码如下!

class Solution {
    public int searchInsert(int[] nums, int target) {
        int low=0;
        int high=nums.length-1;
        int mid=0;
        while(low<=high){
            mid=(low+high)/2;
            if(target>nums[mid]){
                low=mid+1;
            }else if(target<nums[mid]){
                high=mid-1;
            }else{
                break;
            }
        }
        if(target==nums[mid]) return mid;
        else if(target>nums[mid]) return mid+1;
        else if(target<nums[mid]) return mid;
        return mid;
    }
}

你可能感兴趣的:(刷题日记,leetcode,算法)