搜索插入位置

搜索插入位置

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

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

示例 1:

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

示例 2:

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

示例 3:

输入: [1,3,5,6], 7
输出: 4

示例 4:

输入: [1,3,5,6], 0
输出: 0

官方思路是：
设pos是要求的索引值，有2种情况，

• 一种是存在这个数，这时target = nums[pos]，pos是所要求的索引
• 另一种是不存在这个数，nums[pos - 1] < target < nums[pos]，pos是所要求的索引

合并一下就是nums[pos - 1] < target <= nums[pos]，就是要找序列中第一个大于等于target的数

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0;
		int right = nums.length - 1;
		int pos = nums.length;
		int mid;
		while(left <= right) {
			mid = left + (right - left) >> 1;
			if(nums[mid] >= target) {
				pos = mid;
				right = mid - 1;
			} else {
				left = mid + 1;
			}
		}
		return pos;
    }
}

我觉得在找到时就可以返回，while的if如下，

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

但是我觉得官方代码比这个好理解

思路2：
我还发现了规律，先二分法找这个target，如果找不到，此时左指针指向的位置就是所找索引

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

官方链接：https://leetcode-cn.com/problems/search-insert-position/solution/sou-suo-cha-ru-wei-zhi-by-leetcode-solution/