1. 程式人生 > 其它 >LeetCode No35. 搜尋插入位置

LeetCode No35. 搜尋插入位置

題目

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

請必須使用時間複雜度為 O(log n) 的演算法。

示例 1:

輸入: nums = [1,3,5,6], target = 5
輸出: 2

示例 2:

輸入: nums = [1,3,5,6], target = 2
輸出: 1

示例 3:

輸入: nums = [1,3,5,6], target = 7
輸出: 4

提示:

1 <= nums.length <= 10^4
-10^4 <= nums[i] <= 10^4
nums 為 無重複元素 的 升序 排列陣列
-10^4 <= target <= 10^4

思路

有序陣列找值的下標,簡單二分。

AC程式碼

點選檢視程式碼
class Solution {

    public int bSearch(int[] nums, int low, int high, int target) {
        int mid = (low+high)/2;
        if( low>=high ) {
            return high;
        }
        if( nums[mid] == target ) {
            return mid;
        } else if( nums[mid]>target ) {
            high = mid;
        } else {
            low = mid + 1;
        }
        return bSearch(nums, low, high, target);
    }

    public int searchInsert(int[] nums, int target) {
        int len = nums.length;
        int index = bSearch(nums, 0, len, target);
        return index;
    }
}