【Leetcode-565】数组嵌套

LeetCode 每日一题

题目链接

https://leetcode.cn/problems/array-nesting/

题目描述

数组嵌套


索引从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}


来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/array-nesting
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

分析题目：
题目的意思实际上是，以一个索引开始，然后将他的值作为新的索引不断去访问数组元素，直到出现重复元素或者数组元素全部访问为止。
这是一个非常典型的图的遍历问题。

1. 定义一个 visited 数组，保存当前索引的元素是否访问，以任意索引开始（一般以第 0 个），不断访问，直到重复或者结束，记录长度
2. 以剩余任意一个未访问到的元素开始，重复上面步骤，直到所有元素都被访问到位置
3. 返回最大长度

题解

class Solution {
  
   public static int arrayNesting(int[] nums) {

        int max = 0;
        boolean[] visited = new boolean[nums.length];
        for (int i = 0; i < nums.length; i++) {
            int count = 0;
            while (!visited[i]) {
                visited[i] = true;
                ++count;
                i = nums[i];
            }
            max = Math.max(max, count);
        }
        return max;
    }
}

运行结果