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出现重复的元素。

提示：

N是[1, 20,000]之间的整数。

A中不含有重复的元素。

A中的元素大小在[0, N-1]之间。

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²) 在①处嵌套查找次数与A大小n相关

空间复杂度：O(n) 需要额外用到一个unordered_set

2.图

题目表明A中不存在重复的元素，也就是说可以把嵌套查找A[A[i]]...看作对出度和入度为一的有向图的遍历，且A所遍历的结果一定是一个或多个环。如下图所示，从任意一个环内的元素开始可以遍历整个环，所以我们嵌套查找时访问到的元素就可以打上标记，之后A遍历到该元素时可以跳过；

class Solution {

public:

    int arrayNesting(vector<int>& nums) {

        int max_size = 0,N = nums.size();

        for(int i=0;i<N;++i){    //①

            int count = 0;

            int j = i;

            while(nums[j] < N){    //②

                ++count;

                max_size = max(max_size, count);

                int tmp  = j;

                j = nums[j];

                nums[tmp] = N;

            }

        }

        return max_size;

    }

};

时间复杂度：O(n) 对A中每个元素最多遍历2n次，即 ①②两处各访问一次

空间复杂度：O(1) 不占用额外空间

参考

https://leetcode.cn/problems/array-nesting/solution/shu-zu-qian-tao-by-leetcode-solution-7ur3/

巴特西

LEETCODE 数组嵌套

题目:数组嵌套

提示：

1.暴力解法

2.图

参考

最新文章

热门文章