魔术索引。 在数组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,即数组中可能包含重复元素的版本
分析:
方法1:朴素解法
最简单粗暴的方法就是从头到尾依次遍历,返回第一个索引等于值的索引。
时间复杂度:O(n)
空间复杂度:O(1)
class Solution {
public int findMagicIndex(int[] nums) {
//遍历
for(int i = 0; i < nums.length; i++){
//索引等于值
if(nums[i] == i){
return i;
}
}
return -1;
}
}
方法2:分治算法+DFS+剪枝
因为数组中的数有重复情况,所以不能直接用二分法进行计算,但是可以用二分的思想进行优化。因为要找最小的等于值的索引,因此该索引很大可能在二分点的左边,因此我们可以用深度遍历对左边进行遍历,若返回的不为 -1,则该索引一定为目标索引,若为 -1,就判断当前索引是否等于值,若还为 -1,就继续遍历右边,直到找到索引,或者没有。
时间复杂度:O(n) 最坏情况和线性遍历一样
空间复杂度:O(n)
class Solution {
public int findMagicIndex(int[] nums) {
return recur(nums, 0, nums.length-1);
}
public int recur(int[] nums, int i, int j){
if(i > j){
return -1;
}
//二分点
int mid = (i + j) / 2;
//遍历左边
int left = recur(nums, i, mid-1);
//左边有,返回该索引
if(left != -1){
return left;
}
//当前索引等于值
if(mid == nums[mid]){
return mid;
}
//遍历右边
return recur(nums, mid + 1, j);
}
}
方法3:跳跃遍历
因为数组中的元素都是递增的,且一定小于等于最大索引,所以当前索引的值一定存在和值一样的索引,若该值比当前索引大,那么该值和索引之前的值,不是重复就是值大于索引,因此我们可以直接跳跃到和该值一样的索引进行遍历,直到找到目标索引或遍历完成。
时间复杂度:O(n) 最坏情况和线性遍历一样
空间复杂度:O(1)
class Solution {
public int findMagicIndex(int[] nums) {
//遍历,每次选下一个索引和当前值中最大的进行跳跃
for(int i = 0; i < nums.length; i = Math.max(i+1, nums[i])){
//找到最小索引
if(i == nums[i]){
return i;
}
}
return -1;
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/magic-index-lcci