1. 程式人生 > >leetcode 搜尋插入位置

leetcode 搜尋插入位置

題目描述:

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

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

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

 解題思路:

小心處理極小值和極大值。然後其他情況,只需要找到第一個大於target的位置,返回該位置即可

C++程式碼:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        
        int len = nums.size();
        if (target <= nums[0]) return 0; // target比陣列中最小的還要小,或相等,則插在0位置
        
        for (int i = 0; i < len; i++){   // 遍歷
            if (nums[i] == target) return i; // 存在
            else if (target < nums[i]) return i; // 陣列中不存在,找到第一個大於target的位置,返回該位置
            
        }
        
        return len; // 執行到這,說明target大於陣列中所有值,插在len位置
        

    }
};