题目简介:
魔术索引。 在数组A[0...n-1]
中,有所谓的魔术索引,满足条件A[i] = i
。给定一个有序整数数组,编写一种方法找出魔术索引,若有的话,在数组A中找出一个魔术索引,如果没有,则返回-1。若有多个魔术索引,返回索引值最小的一个。
示例1:
1 | 输入:nums = [0, 2, 3, 4, 5] |
示例2:
1 | 输入:nums = [1, 1, 1] |
说明:
- nums长度在[1, 1000000]之间
- 此题为原书中的 Follow-up,即数组中可能包含重复元素的版本
思路:
首先最简单的就是循环遍历一次数组即可。
为了优化,我们可以试想这么一种情况,nums[i] > i + 1
,如nums[0] = 2 > 1
,由于是有序数组,所以nums[1 ... n] >= 2
,为了减少遍历次数,我们可以把i += nums[i] - 1
( - 1的目的是抵消循环中的i++
)。如上面的例子中,下一次循环便从nums[2]
开始。
tip:
- 数组中存在重复元素,所以
i += nums[i] - 1
。
代码如下:
1 | class Solution { |