二分陷阱

二分陷阱

preface

说起二分法，其实很多人会想到 $O(logn)$ 的时间复杂度，但是其实不然，有些在极端条件下依然会达到 $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

题目

[官解](