题目
测试数据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数组: