leetcode #565. 数组嵌套

题目描述：

索引从0开始长度为N的数组A，包含0到N - 1的所有整数。找到最大的集合S并返回其大小，其中 S[i] = {A[i], A[A[i]], A[A[A[i]]], … }且遵守以下的规则。

假设选择索引为i的元素A[i]为S的第一个元素，S的下一个元素应该是A[A[i]]，之后是A[A[A[i]]]… 以此类推，不断添加直到S出现重复的元素。

思路：

1、可以发现，所有符合的元素都可以构成一个环，无论从哪个元素开始，最终都会回到这个元素；
2、构建一个辅助队列，如果已经用到这个元素的话，标记一下，避免重复计算

代码：

class Solution:
    def arrayNesting(self, nums):
        anslist = []
        fuzhu = [None]*(len(nums))
        for i in range(len(nums)):
            split = []
            while nums[i] not in split and fuzhu[i] == None:
                fuzhu[i] = 1
                i = nums[i]
                split.append(i)
            anslist.append(split)
        return max(len(x) for x in anslist)