天天看點

Leetcode 35. 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      

解題思路:

排序數組,使用二分查找。

時間複雜度O(logn),空間複雜度O(1)。

代碼實作:

class Solution {
    public int searchInsert(int[] nums, int target) {
        int start = 0, end = nums.length - 1;
        while (start <= end) {
            int mid = start + (end - start) / 2;
            if (nums[mid] > target) {
                end = mid - 1;
            } else if (nums[mid] == target) {
                return mid;
            } else {
                start = mid + 1;
            }
        }
        return start;
    }
}