面试题 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

分析： 二分法

方法1： 二分查找

class Solution:
    def findMagicIndex(self, nums) -> int:
        def binary_search(left, right):
            mid = (left+right)//2
            if mid == left:
                if nums[left] == left:
                    return left
                elif nums[right] == right:
                    return right
                else:
                    return -1
            if nums[right] >= left and nums[left] <= right:
                if nums[mid] == mid:
                    return binary_search(left, mid)
                else:
                    left_res = binary_search(left, mid)
                    right_res = binary_search(mid, right)
                    if left_res != -1:
                        return left_res
                    return right_res

            else:
                return -1
        return binary_search(0, len(nums)-1)

方法2： 直接遍历，但是取max(i, nums[i])作为下一个遍历的起点，此方法只适合单调递增的数组

class Solution:
    def findMagicIndex(self, nums) -> int:
        res = -1
        i = 0
        while i < len(nums):
            if i == nums[i]:
                return i
            i = max(i+1, nums[i])
        return -1