蓝桥杯小朋友崇拜圈（递推，dfs找环，优化）

题目

测试数据1：
9
3 4 2 5 3 8 4 6 9
测试数据2：
30
22 28 16 6 27 21 30 1 29 10 9 14 24 11 7 2 8 5 26 4 12 3 25 18 20 19 23 17 13 15

代码

#include<bits/stdc++.h> 
using namespace std;
int par[100010];
int dep[100010];//即可存储深度，也可表示是否被访问过 
int re[100010];// re[num] : 从num出发得到的环元素的个数
int dfs(int num,int depth)//返回值为 从num出发得到的环元素的个数
{
	if(re[num]!=0) return re[num];
	if(dep[num]!=0)
	{
		return depth-dep[num];
	}	
	else
	{
		dep[num]=depth;
		re[num]=dfs(par[num],depth+1);
		dep[num]=0;//这里没有for循环，其回溯目的是为了清空dep数组 
		return re[num];
	}
}
int main()
{
	int N;
	cin>>N;
	for(int i=1;i<=N;i++)
	{
		int temp;
		cin>>temp;
		par[i]=temp;
	}
	for(int i=1;i<=N;i++)
	{
		re[i]=dfs(i,1);
	}
	int result=-1;
	for(int i=1;i<=N;i++)
	{
		cout<<re[i]<<" ";
		result=max(result,re[i]);
	}
	cout<<"结果为"<<result; 
}

找环关键：
return depth-dep[num]

re[num] : 从num出发得到的环元素的个数
这里使用re数组进行优化,如果说dfs到 曾经访问过的小朋友 ，那么两个小朋友的re一定是一样的。
如下图：
re数组：