当C++遇上LeetCode——二分法

题目引入

题目:33.搜索旋转排序数组

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

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

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

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

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

示例 1:

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

示例 2:

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

我的题解

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;
            else if (nums[mid] < nums[right]) {
                if (nums[mid] < target && nums[right] >= target) left = mid + 1;
                else right = mid - 1;
            } else {
                if (nums[left] <= target && nums[mid] > target) right = mid - 1;
                else left = mid + 1;
            }
        }
        return -1;
    }
};

简单分析

• 对于二分，一个关键字就是复杂度 O(log n) ，此外对于二分要记住一个标准模版，这里写int mid = left + (right - left) / 2;是为了防止出现left+right过大，超过范围
• 对与这道题关键是思考清楚选左还是右

类似题目

• 35. 搜索插入位置
• 34. 在排序数组中查找元素的第一个和最后一个位置.md
• 278. 第一个错误的版本

后记

• 二分法关键在与很多时候不是让你搜索出固定数，而是看趋势，此时你的mid怎么偏就很有讲究了