剑指offer 专项突破版 68、查找插入位置

题目链接

思路

• 二分查找主要就是判断何时该分，对于此题的要求
• 如果nums[mid] < target,我们要找的位置一定还在mid右边
• 如果nums[mid] > target，还要再判断，如果与此同时nums[mid - 1] < target，那么说明不存在这样的target，此时的mid就是待插入的位置，但是既然提到了mid-1，就必须在nums[mid-1]前先判断mid是否为0

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0, right = nums.length - 1, result = nums.length;

        while (left <= right) {
            int mid = left + ((right - left) >> 1);

            if(nums[mid] == target)
                return mid;
            if(nums[mid] > target){
                if(0 == mid || nums[mid - 1] < target)
                    return mid;
                right = mid - 1;
            }else
                left = mid + 1;
        }
        return result;
    }
}

Go版本

func searchInsert(nums []int, target int) int {
	left, right := 0, len(nums)-1

	for left <= right {
		mid := left + ((right - left) >> 1)
		if target == nums[mid] {
			return mid
		} else if target > nums[mid] {
			left = mid + 1
		} else {
			// 如果if不成立，说明mid对应的不是第一个大于target的数，所以可以修改right为mid-1
			if mid == 0 || nums[mid-1] < target {
				return mid
			}
			right = mid - 1
		}
	}
	// 运行到这里说明数组中没有比target大的元素
	return len(nums)
}

注意

• 注意二分的标准模板，while条件是left 小于等于 right，因为如果不是小于等于，那么当left=right时就没法判断这个元素是否和target相等了
• 在本题中关键在于target < nums[mid]的处理，因为找的是第一个大于等于target的元素，所以这个时候不知道应不应该让right = mid-1，那么这就需要对mid做判断，如果mid是0或者是nums[mid]比target小，就说明nums[mid]是第一个比target大的数，直接返回mid即可，反之则说明不是，那就可以放心的让right = mid-1
• 当循环结束还没退出时说明数组没有大于等于target数，所以直接返回数组长度~