题目描述:
索引从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}
注意:
N是[1, 20,000]之间的整数。
A中不含有重复的元素。
A中的元素大小在[0, N-1]之间。
比较简单,顺序进行遍历,由于能够形成环,那么环内的所有元素的长度都是一致的,因此只需要计算其中一个即可,isvisited表示的是计算过的,防止重复计算,从而导致超时
class Solution {
public int arrayNesting(int[] nums) {
int size = 0;
Set<Integer> isvisted = new HashSet<>();
Set<Integer> set = new HashSet<>();
for (int i = 0; i < nums.length; i++) {
if(isvisted.contains(i)){
continue;
}
set.clear();
set.add(nums[i]);
isvisted.add(nums[i]);
int tem = nums[i];
while (!set.contains(nums[tem])) {
set.add(nums[tem]);
isvisted.add(nums[tem]);
tem = nums[tem];
}
size = Math.max(size, set.size());
}
return size;
}
}
不使用set的话可能快一点,排名靠前的代码
每次遍历后都将其置为-1
class Solution {
public int arrayNesting(int[] nums) {
int max = 0;
// 顺序遍历,并将已经遍历过的节点为 -1 ,避免重复遍历。
// 关键点:能够顺序的原因是,嵌套数组,必然形成一个环。从任一一个节点进入,整个环的大小是一致的。
for (int i = 0; i < nums.length; i++) {
// 跳过,如果已经访问
if (nums[i] == -1) {
continue;
}
int counts = 0;
// 从当前节点开始,不断遍历
for (int j = i; nums[j] != -1;) {
counts++;
int tmp = nums[j];
nums[j] = -1;
j = tmp;
}
// 计算最大
max = Math.max(max, counts);
}
return max;
}
}