记忆化搜索10分求助

#include<bits/stdc++.h>
using namespace std;

int n;
int nxt[1000005],dp[1000005];
int vis[1000005];
int isr[1000005];

void dfs(int x,int step)
{
	if(vis[x])
	{
		if(isr[x]) return;
		isr[x]=step-vis[x];
	}
	vis[x]=step;
	dfs(nxt[x],step+1);
	return ;
}

int dfs2(int x)
{
	if(dp[x]) return dp[x];
	if(isr[x]) return isr[x];
	return dp[x]=dfs2(nxt[x])+1;
}

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>nxt[i];
	for(int i=1;i<=n;i++)
	{
		if(!vis[i]) dfs(i,1);
	}
	for(int i=1;i<=n;i++)
	{
		cout<<dfs2(i)<<endl;
	}
	return 0;
}