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:

输入: 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}
而剩余的{1, 4}和{3}同样构成循环，但不是最长。

链接：https://leetcode.com/problems/array-nesting/

解题思路：

1. 将嵌套数组看作环状链表，此题立马变的醒目。
2. 不难看出，这是由若干结点构成的多个环状链表，让你求出最长的那个。
3. 明确一点，这些链表必定都是环状链表（单结点，指向自身的也算）
4. 既然是环状链表，那么从哪个结点出发都一样，最后都能得到相同的length。
5. 剪枝：将遍历过的结点置为-1， 以免重复遍历。
6. 最终原数组的所有元素都会是-1

我的代码：

class Solution(object):
    def arrayNesting(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        # 参考457：环形数组循环
        res = 0
        l = len(nums)
        for i in range(l):
            if nums[i] == -1:
                continue
            # 环状链表起点
            j = i
            cur = 0
            while nums[j] >= 0:
                cur += 1
                temp = nums[j]
                nums[j] = -1   # 剪枝
                j = temp
            # 更新最大长度
            res = max(res, cur)
        return res

Time: O(N)
Space： O(1)
难度：中等
标签：链表