C++의 배열 중첩

<시간/>

0에서 N-1까지의 모든 정수를 포함하는 길이가 N인 0 인덱스 배열 A가 있다고 가정합니다. S[i] ={A[i], A[A[i]], A[A[A[i]]], ... } 집합 S의 가장 긴 길이를 찾아서 반환해야 합니다. 아래 규칙. 이제 S의 첫 번째 요소가 index =i의 요소 A[i] 선택으로 시작하고 S의 다음 요소는 A[A[i]]이어야 하고 A[A[A[i]]]… 그 유추에 의해, 중복 요소가 S에서 발생하기 직전에 추가를 중지합니다. 따라서 배열이 A =[5,4,0,3,1,6,2]와 같으면 출력은 A[ 0] =5, A[1] =4, A[2] =0, A[3] =3, A[4] =1, A[5] =6, 마지막으로 A[6] =2.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

dfs라는 함수를 만듭니다. 이것은 노드, arr 배열, v 배열 및 방문한 세트를 취합니다. dfs 배열에서 다음을 수행하십시오 -
노드를 방문하면 반환
v에 노드 삽입, 노드를 방문한 것으로 표시
dfs(arr[노드], arr, v, 방문)
메인 방법에서 다음을 수행하십시오. -
ret :=0, n :=숫자의 크기. 방문이라는 세트를 만드세요
0 ~ n – 1 범위의 i에 대해
- 배열 만들기 v
- nums[i]를 방문하지 않으면 dfs(nums[i], nums, v, Visited)
- ret :=ret의 최대값과 v의 크기
반환

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   void dfs(int node, vector <int>& arr, vector <int>& v, set <int>& visited){
      if(visited.count(node)) return;
      v.push_back(node);
      visited.insert(node);
      dfs(arr[node], arr, v, visited);
   }
   int arrayNesting(vector<int>& nums) {
      int ret = 0;
      int n = nums.size();
      set <int> visited;
      for(int i = 0; i < n; i++){
         vector <int> v;
         if(!visited.count(nums[i]))dfs(nums[i], nums, v, visited);
         ret = max(ret, (int)v.size());
      }
      return ret;
   }
};
main(){
   vector<int> v = {5,4,0,3,1,6,2};
   Solution ob;
   cout << (ob.arrayNesting(v));
}

입력

[5,4,0,3,1,6,2]