leetcode 每日一题(3)——查找给定数字索引

  首发于个人博客

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

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

示例 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：此题比较简单，初读题目感觉是个查找算法。但是有一个非常明确的区别就是可能位于两个数之间，这个就是需要额外注意补充的逻辑。

下面用二分查找来实现一下思路。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        return searchInsertReal(nums,0,nums.size() - 1,target);
    }

    int searchInsertReal(vector<int>& nums,int startIndex,int endIndex,int target){
        unsigned long size = endIndex - startIndex + 1;
        if (size == 0){
            return -1;
        }
        int centerIndex = size / 2 + startIndex;
        int centerTarget = nums[centerIndex];
        if (centerTarget == target){
            //找到了，直接返回结果
            return centerIndex;
        } else if(centerTarget > target){
            if (size > 1){//22222222222
                //目标比中间位子数值小，比较前一个，验证是不是在比前一个小。
                int preValue = nums[centerIndex - 1];
                if (preValue < target){
                    //小于前一个值，则返回结果
                    return centerIndex;
                }else{
                    //继续查找，因为已经验证了比前一个还小，则位子可以减1。
                    return searchInsertReal(nums,startIndex,centerIndex - 1,target);
                }
            } else{
                return centerIndex;
            }
        } else{
            //centerTarget < target 
            if (size > 2){ //11111111
                //目标比中间位子数值大，比较后一个，验证是不是比后一个大
                int later = nums[centerIndex + 1];
                if (later > target){
                    //比后一个小，返回结果
                    return centerIndex + 1;
                } else{
                    //继续查找
                    return searchInsertReal(nums,centerIndex + 1,endIndex,target);
                }
            } else{
                return centerIndex + 1;
            }
        }
    }


};

上面代码 111111处注释，是size > 2 , 222222处注释是size > 1.

这是因为我们 cneterIndex是 size / 2.如果是size是2的话，centerIndex =1已经是最后一个元素了。所以size最少是3才会有后一个，最少是2才会有前一个。