正文
第四题:[USACO08DEC]Trick or Treat on the Farm
其实这道题已经规定了每个点的出度只有1,那么就很好做了。
每个点建立一个时间戳数组d[i],初始化为0,当一个点从自己出发,回到了自己,那么就等于,回到的时间戳-1就是所在环的大小。
如果一个点不在环中,那么,他一定可以遍历到一个在环的点,用到这的时间戳+这个环的大小即可。
代码<long long long>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;
int n;
int t[100010];
struct node{
int x,y,last;
}s[100010];
int tt=0;
queue<int> f;
void bfs(int x){
f.push(x);
s[x].x=tt;
s[x].y=1;
s[x].last=0;
while(!f.empty()){
int x=f.front();
f.pop();
int y=t[x];
//printf("%d %d %d %d\n",x,t[x],s[x].x,s[x].y);
if(s[y].x==s[x].x){
s[y].y=s[x].y+1-s[y].y;
for(int i=x;i!=0 && i!=y;i=s[i].last)
s[i].y=s[y].y;
int now=0;
for(int j=y;j!=0;j=s[j].last)
s[j].y=s[y].y+(now++);
return ;
}
else if(s[y].x!=0)
{
int now=1;
for(int i=x;i!=0;i=s[i].last)
s[i].y=s[y].y+(now++);
return ;
}
else {
s[y].x=s[x].x;
s[y].y=s[x].y+1;
s[y].last=x;
f.push(y);
}
}
return ;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&t[i]);
for(int i=1;i<=n;i++){
if(s[i].x==0)
{
tt++;
bfs(i);
}
}
for(int i=1;i<=n;i++)
printf("%d\n",s[i].y);
}