LeetCode算法 —— 搜索旋转排序数组

题目 :
假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组?[0,1,2,4,5,6,7]?可能变为?[4,5,6,7,0,1,2]?)。

搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回?-1?。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是?O(log?n) 级别。

二分探索(O(LogN))

#include <iostream>
#include <stack>
#include <algorithm>
#include <vector>

using namespace std;

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        if (!n) return -1;      // 判断是否有数据

        if (n == 1) return nums[0] == target ? 0 : -1;  // 只有一个数据的情况 

        int l = 0, r = n - 1;
        while (l <= r) {
            int mid = (l + r) / 2;
            if (nums[mid] == target) return mid;        // 判断区间中间的那个数是否是目标值

            // 左半部分是有序的
            if (nums[l] <= nums[mid]) {
                // 判断是否在左半部分
                if (nums[l] <= target && target < nums[mid]) {
                    r = mid - 1;
                }
                else {
                    l = mid + 1;
                }
            }
            else {  // 右半部分是有序的
                // 判断是否在右半部分
                if (nums[mid] < target && target <= nums[n - 1]) {
                    l = mid + 1;
                }
                else {
                    r = mid - 1;
                }
            }
        }
        return -1;
    }
};
 

int main() {
    vector<int> vec{ 4,5,6,7,1,2,3 };
    cout << (new Solution())->search(vec, 3);
	 
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_42100963/article/details/107618321