王廷瑋|數位醫療|智慧醫療: 33. Search in Rotated Sorted Array WFU

2024年7月3日 星期三

33. Search in Rotated Sorted Array

33. Search in Rotated Sorted Array


有一個按升序排序的整數數組 nums(具有不同的值)。

在傳遞給你的函數之前,nums 可能在未知的旋轉點 k 處進行了旋轉(1 <= k < nums.length),使得結果數組為 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]](0 索引)。例如,[0,1,2,4,5,6,7] 可能在旋轉點索引 3 處進行旋轉,變為 [4,5,6,7,0,1,2]。

給定可能旋轉後的數組 nums 和一個整數 target,如果 target 在 nums 中,則返回 target 的索引;如果不在 nums 中,則返回 -1。

你必須設計一個時間複雜度為 O(log n) 的算法。

範例 1:

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


Python


from typing import List

class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1

while left <= right:
mid = (left + right) // 2
if nums[mid] == target:
return mid

# Determine which part is sorted
if nums[left] <= nums[mid]:
# Left part is sorted
if nums[left] <= target < nums[mid]:
right = mid - 1
else:
left = mid + 1
else:
# Right part is sorted
if nums[mid] < target <= nums[right]:
left = mid + 1
else:
right = mid - 1

return -1

16.81MB, 48ms


C++


#include <vector>

using namespace std;

class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;

while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
return mid;
}

// Determine which part is sorted
if (nums[left] <= nums[mid]) {
// Left part is sorted
if (nums[left] <= target && target < nums[mid]) {
right = mid - 1;
} else {
left = mid + 1;
}
} else {
// Right part is sorted
if (nums[mid] < target && target <= nums[right]) {
left = mid + 1;
} else {
right = mid - 1;
}
}
}

return -1;
}
};

13.35MB, 4ms


Javascript


/**
* @param {number[]} nums
* @param {number} target
* @return {number}
*/
var search = function(nums, target) {
let left = 0;
let right = nums.length - 1;

while (left <= right) {
const mid = Math.floor((left + right) / 2);
if (nums[mid] === target) {
return mid;
}

// Determine which part is sorted
if (nums[left] <= nums[mid]) {
// Left part is sorted
if (nums[left] <= target && target < nums[mid]) {
right = mid - 1;
} else {
left = mid + 1;
}
} else {
// Right part is sorted
if (nums[mid] < target && target <= nums[right]) {
left = mid + 1;
} else {
right = mid - 1;
}
}
}

return -1;
};

48.28MB, 68ms