二分陷阱
preface
说起二分法,其实很多人会想到 O ( l o g n ) O(logn) O(logn) 的时间复杂度,但是其实不然,有些在极端条件下依然会达到 O ( n ) O(n) O(n) 的复杂度。
Solution
面试题 08.03. 魔术索引
魔术索引。 在数组A[0...n-1]中,有所谓的魔术索引,满足条件A[i] = i。给定一个有序整数数组,编写一种方法找出魔术索引,若有的话,在数组A中找出一个魔术索引,如果没有,则返回-1。若有多个魔术索引,返回索引值最小的一个。
示例1:
输入:nums = [0, 2, 3, 4, 5]
输出:0
说明: 0下标的元素为0
示例2:
输入:nums = [1, 1, 1]
输出:1
说明:
nums长度在[1, 1000000]之间
此题为原书中的 Follow-up,即数组中可能包含重复元素的版本
很简单,如果忽略有序条件,直接遍历一遍即可
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int findMagicIndex(vector<int>& nums) {
auto i = 0;
while (i < static_cast<int>(nums.size()))
{
if (i == nums[i])
return i;
i = max(i + 1, nums[i]);
}
return -1;
}
};
int main()
{
vector<int> vec{ 1, 1, 1 };
Solution s;
cout << s.findMagicIndex(vec);
return 0;
}
当然了,因为当初的遍历太慢了,可以考虑跳着来。
接下来就是重点,二分。
主要是题目知道了有重复,我们就不得不一直往左边去遍历,也导致了其实这个二分需要剪枝,我已经找到最小的,就没有必要再往右边找了。
class Solution {
private:
static void binary_search(vector<int>& nums, int left, int right, int& ret) {
if (ret != -1 && ret < left) return;
if (left <= right) {
const auto mid = left + ((right - left) >> 1);
if (nums[mid] == mid) {
if (ret == -1) ret = mid;
else if (mid < ret) ret = mid;
binary_search(nums, left, mid - 1, ret);
}
else {
binary_search(nums, left, mid - 1, ret);
binary_search(nums, mid + 1, right, ret);
}
}
}
public:
static int findMagicIndex(vector<int>& nums) {
auto ret = -1;
binary_search(nums, 0, nums.size() - 1, ret);
return ret;
}
};
int main()
{
vector<int> vec{ 1, 1, 1 };
Solution s;
cout << s.findMagicIndex(vec);
return 0;
}
这样就能找到最小的情况下,我们能比直接遍历更加快。
reference
[官解](