LeetCode 565 数组嵌套[dfs 记忆化] HERODING的LeetCode之路

Posted 2022-07-18 HERODING23

解题思路：
首先要弄清楚数组的特点，在这个数组中，所有节点的出度为0，入度为0，这就表明数组是由各个节点构成的多个环组成（也可能就一个环，但是肯定是环），所以我们只需要找到最大的那个环长度即可，单纯的暴力递归会超时，最好的方法是用记忆化，将访问过的节点记录下来，再次访问直接跳过（因为是同一个环），如果当前环的长度超过数组的一半，那么该环的长度一定是最长的，直接返回即可，代码如下：

class Solution 
public:
    int arrayNesting(vector<int>& nums) 
        int n = nums.size();
        vector<bool> visited(n, false);
        int maxLen = 1;
        for(int i = 0; i < n; i ++) 
            if(maxLen > n / 2) return maxLen;
            if(visited[nums[i]]) continue;
            int curLen = 1;
            visited[nums[i]] = true;
            int cur = nums[i];
            while(!visited[nums[cur]]) 
                visited[nums[cur]] = true;
                curLen ++;
                cur = nums[cur];
            
            cout << endl;
            maxLen = max(maxLen, curLen);
        
        return maxLen;
    
;