【leetcode】面试题 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，即数组中可能包含重复元素的版本

思路：由于题目中没有说是什么顺序，因此最简单的就是循环遍历整个数组，然后通过一个TreeSet来进行记录，最后取该set中最小的那个。但是后来看了题解发现一个现象：仍然是从头开始遍历，如果当前位置上的值比下标要大，那么把当前位置的值作为下标的话，新下标对应的值肯定是大于或者等于（如果一直是重复的值）当前位置的下标。因此在这种情况下，可以直接跳到新下标的位置进行后面的遍历即可。如果当前位置的值比下标要小，那么直接往后移动一位即可，代码如下：

class Solution {
    public int findMagicIndex(int[] nums) {
        if(nums == null || nums.length == 0){
            return -1;
        }
        int flag = 0;
        while(flag < nums.length){
            if(nums[flag] == flag){
                return flag;
            }else if(nums[flag] > flag){
                // 说明在flag位置之前的都不可能是符合条件的
                // 因为nums[nums[flag]] >= nums[flag]
                flag = nums[flag];
            }else{
                flag ++;
            }
        }
        return -1;
        //------------not the best---------------//
        // TreeSet<Integer> set = new TreeSet<>();
        // for(int i = 0; i< nums.length ; i++){
        //     if(nums[i] == i){
        //         set.add(i);
        //     }
        // }
        // return set.isEmpty() ? -1 : (int)set.first();
    }
}