P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm
分析:
这棵树上有且仅有一个环
两种情况:
1.讨论一个点在环上,如果在则答案与它指向点相同,
2.不在就等于它指向点答案+1,具体就直接大力dfs,
如何求环长度:
终点深度-起点深度=终点起点距离
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
const int maxn=;
int n;
int cnt=;
int a[maxn];
int h[maxn];
int ans[maxn];
int read() //快读
{
int x=,f=;char c=getchar();
while(!isdigit(c)){if(c=='-')f=-;c=getchar();}
while(isdigit(c)){x=x*+c-'';c=getchar();}
return x*f;
}
void dfs(int i,int deep)
{
h[i]=deep; //该点深度
if(ans[a[i]]) ans[i]=ans[a[i]]+; //若指向点有答案,而自己无答案,则自己不再环上。要+1
else if(h[a[i]]) //若形成环
{
ans[i]=(h[i]-h[a[i]]+); //刷新该点答案
int k=a[i];
while(i!=k) //刷新环内所有答案 ,直到回到原点
{
ans[k]=ans[i];
k=a[k];
}
}
else
{
dfs(a[i],deep+);
if(!ans[i]) ans[i]=ans[a[i]]+; //自身肯定不再环上,所以+1
}
}
void solve()
{
for (int i=;i<=n;i++)
{
if(!ans[i]) dfs(i,); //若此点未有涉及(无答案),就dfs
}
for (int i=;i<=n;i++)
printf("%d\n",ans[i]);
}
void work()
{
n=read();
for(int i=;i<=n;i++)
{
a[i]=read();
}
solve();
}
int main()
{
work();
return ;
}