1. 程式人生 > 其它 >LeetCode: 704.二分查詢

LeetCode: 704.二分查詢

給定一個n個元素有序的(升序)整型陣列nums 和一個目標值target ,寫一個函式搜尋nums中的 target,如果目標值存在返回下標,否則返回 -1。

示例 1:

輸入: nums = [-1,0,3,5,9,12], target = 9
輸出: 4
解釋: 9 出現在 nums 中並且下標為 4
示例2:

輸入: nums = [-1,0,3,5,9,12], target = 2
輸出: -1
解釋: 2 不存在 nums 中因此返回 -1

提示:

你可以假設 nums中的所有元素是不重複的。
n將在[1, 10000]之間。
nums的每個元素都將在[-9999, 9999]之間。

分析

如果目標值等於中間元素,則找到目標值。
如果目標值較小,繼續在左側搜尋。
如果目標值較大,則繼續在右側搜尋。

具體過程:

初始化指標 left = 0, right = n - 1。
當 left <= right:
比較中間元素 nums[mid] 和目標值 target 。
如果 target = nums[mid],返回 mid。

如果 nums[mid] < traget,則在右側繼續搜尋 left = mid + 1。
如果 nums[mid] > traget,則在左側繼續搜尋 right = mid - 1。

Python版本:

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left 
= 0 right = len(nums) -1 while left <= right: mid = left + (right - left) //2 if nums[mid] < target: left = mid + 1 elif nums[mid] > target: right = mid -1 else: return mid return
-1

Java版本:

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

複雜度分析

  • 時間複雜度:O(logN)。
  • 空間複雜度:O(1)。

參考資料:

704. 二分查詢 - 力扣(LeetCode) (leetcode-cn.com)