LeetCode 565. Array Nesting

問題

長度爲N的零索引數組A包含從0到N-1的所有整數。查找並返回集合S的最長長度，其中S[i] = {A[i]， A[A[i]]， A[A[A]]， A[A[i]]，…服從以下規則。假設S中的第一個元素以index = i的元素A[i]的選擇開始，那麼S中的下一個元素應該是A[A[i]]，然後是A[A[A[i]]……

輸入: A = [5,4,0,3,1,6,2]
輸出: 4
解釋: 
A[0] = 5, A[1] = 4, A[2] = 0, A[3] = 3, A[4] = 1, A[5] = 6, A[6] = 2.
最長的S[K]:
S[0] = {A[0], A[5], A[6], A[2]} = {5, 6, 2, 0}

解析

問題很簡單，一句話表示就是：找到數組中最大的“環”。在上面的例子中，A[3]=3，那麼這個環的長度爲1。因此，我們遍歷數組，每次以當前的索引爲開始，下一次又以索引在數組中的值作爲索引，直到最後的取值和最開始的索引值相同，那麼即構成一個“環”。爲了防止重複搜素（一個封閉的環從任意一點開始都可以算是起點），我們新建一個數組標記每個數組的訪問狀態，如果訪問過了，那麼就不再訪問。

Java代碼

    public int arrayNesting(int[] nums) {
        int m = nums.length;
        boolean flag[] = new boolean[m];
        int ans = 0;
        for(int i=0;i<m;i++)if(!flag[i]){
            int start = i;
            flag[start]=true;
            int count=1;
            while(nums[start]!=i){
                start=nums[start];
                flag[start]=true;
                count++;
            }
            ans = Math.max(ans,count);
        }
        return ans;
    }