LeetCode-035. Search Insert Position

1. 题目

Search Insert Position

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
给定已排序的数组和目标值，如果找到目标值，则返回它的下标。如果找不到， 返回它应被插入的位置。假定数组中没有重复项。

2. 分析

1：遍历数组，若nums[i]>=target，则返回i
如果target比nums中任何数都大，则返回数组长度

2：属于变形的二分查找中的其中一种情况
即查找第一个等于或者大于value的元素
由于数组中无重复元素，因此找到时可直接返回

3. 代码

1）

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int len = nums.size();
        if(len <= 0)
            return 0;

        if(target == nums[len - 1])
            return len - 1;
        else if(target > nums[len - 1])
            return len;

        // i < len 而不是 len -1
        for(int i = 0; i < len; ++i)
        {
            // 注意不要把条件写反
            if(nums[i] >= target)
                return i;
        }

        // 注意不要写成 return -1;
        return len;
    }
};

2）

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int len = nums.size();

        if(len <= 0)
            return 0;

        if(target <= nums[0])
            return 0;
        else if(target == nums[len - 1])
            return len - 1;
        else if(target > nums[len - 1])
            return len;

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

        return left;
    }
};

完整源代码放于github