LEETCODE数组嵌套

题目:数组嵌套

索引从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.暴力解法

• 遍历nums数组，遍历到每一个元素时创建一个unordered_set集合 S，再对当前遍历元素按规则 A[A[i]]... 嵌套查找是否元素已经存在于 集合S当中，不存在就插入集合内，存在就结束本次嵌套查找。

class Solution {
public:
    int arrayNesting(vector<int>& nums) {
        size_t max_size = 0;
        int N = nums.size();
        for(int i=0; i<N; ++i){
            unordered_set<int> _set;
            _set.insert(nums[i]);
            max_size = max(max_size, _set.size());  //set插入元素后马上更新集合最大值
            int j = nums[i];
            while(j<N && !_set.count(nums[j]))
            {
                _set.insert(nums[j]);
                j = nums[j];
                max_size = max(max_size, _set.size());
            }
        }
        return max_size;
    }
};

时间复杂度：O(n²)
空间复杂度：O(n)

2.图

• 题目表明A中不存在重复的元素，也就是说可以把嵌套查找A[A[i]]...看作对出度和入度为一的有向图的遍历，且A所遍历的结果一定是一个或多个环。