LeetCode: 704.二分查詢
阿新 • • 發佈:2021-07-18
給定一個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)。
參考資料: