Keshawn_lu's Blog

Leetcode 面试题 08.03. 魔术索引

字数统计: 338阅读时长: 1 min
2020/07/31 Share

题目简介:

魔术索引。 在数组A[0...n-1]中,有所谓的魔术索引,满足条件A[i] = i。给定一个有序整数数组,编写一种方法找出魔术索引,若有的话,在数组A中找出一个魔术索引,如果没有,则返回-1。若有多个魔术索引,返回索引值最小的一个。

示例1:

1
2
3
输入:nums = [0, 2, 3, 4, 5]
输出:0
说明: 0下标的元素为0

示例2:

1
2
输入:nums = [1, 1, 1]
输出:1

说明:

  1. nums长度在[1, 1000000]之间
  2. 此题为原书中的 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
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution {
public:
int findMagicIndex(vector<int>& nums) {

for(int i = 0; i < nums.size(); i++){

if(nums[i] == i)
return i;
if(nums[i] > i + 1)
i += nums[i] - 1;
}

return -1;

}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
  3. 3. 代码如下: