1. 程式人生 > 實用技巧 >LeetCode35-搜尋插入位置

LeetCode35-搜尋插入位置

非商業,LeetCode連結附上:

https://leetcode-cn.com/problems/search-insert-position/

進入正題。

題目:

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

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

示例:

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

程式碼實現:

public int searchInsert(int[] nums, int target) {

        int left = 0, right = nums.length - 1;
        while (left <= right) {
            
            int mid = (right - left) / 2 + left;
            if(nums[mid] == target) {
                return mid;
            } else if(nums[mid] < target) {
                left = mid + 1;
            } else if(nums[mid] > target) {
                right = mid - 1;
            }
        }
        
        return left;
}
//時間複雜度O(logn),空間複雜度O(1)

附註:

參考:https://leetcode-cn.com/problems/search-insert-position/solution/hua-jie-suan-fa-35-sou-suo-cha-ru-wei-zhi-by-guanp/

分析:

根據題目,在“有序陣列”中找目標值,採用“二分查詢”的方式。

題目中要注意的是,當陣列中沒有值和目標值相等時,left位置為插入位置(此時right<left,不再進入while迴圈)。

--End