题目
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
解法1:递归
该题中整形数组为有序数组,属于典型的二分查找应用。递归代码如下:
class Solution {
public:
int search(vector<int>& nums, int target) {
return search(nums, 0, static_cast<int>(nums.size()) - 1, target);
}
private:
int search(vector<int>& nums, int left, int right, int target) {
if (left > right) {
return -1;
}
int mid = left + (right - left) / 2;
if (nums[mid] > target) {
return search(nums, left, mid - 1, target);
}
if (nums[mid] < target) {
return search(nums, mid + 1, right, target);
}
return mid;
}
};
时间复杂度为O(logn)
,空间复杂度为O(logn)
,执行结果如下:
解法2:非递归
二分查找也可以用非递归方法实现,非递归代码如下:
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = static_cast<int>(nums.size()) - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] > target) {
right = mid - 1;
} else if (nums[mid] < target) {
left = mid + 1;
} else {
return mid;
}
}
return -1;
}
};
时间复杂度为O(logn)
,空间复杂度为O(1)
,执行结果如下:
解法3:标准库
对于C++,二分查找已经集成至标准库中,代码如下:
class Solution {
public:
int search(vector<int>& nums, int target) {
auto it = std::lower_bound(nums.begin(), nums.end(), target);
if ((it == nums.end()) || (*it != target)) {
return -1;
}
return (it - nums.begin());
}
};
时间复杂度为O(logn)
,空间复杂度为O(1)
,执行结果如下: