1. 程式人生 > >(LeetCode每日一刷11)搜尋插入位置

(LeetCode每日一刷11)搜尋插入位置

題目描述:

給定一個排序陣列和一個目標值,在陣列中找到目標值,並返回其索引。如果目標值不存在於陣列中,返回它將會被按順序插入的位置。

你可以假設陣列中無重複元素。

 

示例:

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

我提交的程式碼:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) 
    {
        if (nums.empty())
        {
            return 0;
        }
        
        auto len = nums.size();
        for (auto i = 0; i < len; ++i)
        {
            if(nums[i] >= target)
            {
                return i;
            }
            else
            {
                if(i == len - 1)
                {
                    return i + 1;
                }
                
                if(nums[i+1] > target)
                {
                    return i + 1;
                }
            }
                         
        }
    }
    
};