POJ2369【循环节】

题意：
给一个序列，他需要几步就能变成原来的序列。
思路：
那么就是找一下各个循环节(用dfs随便搞了…)，求一下最小公倍数就好了。
贴一发挫code…

//#include <bits/stdc++.h>
#include<iostream>
#include<string.h>
#include<cstdio>
#include<algorithm>
using namespace std;

typedef __int64 LL;

const int N=1e3+10;

int ans;
int ma[N][N];
bool vis[N];
int n;


void dfs(int x)
{
    ans++;
    for(int i=1;i<=n;i++)
    {
        if(ma[x][i]&&!vis[i])
        {
            vis[i]=1;
            dfs(i);
        }
    }
}

int main()
{
    scanf("%d",&n);
    memset(ma,0,sizeof(ma));
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        ma[i][x]=ma[x][i]=1;
    }
    memset(vis,0,sizeof(vis));
    int sum=1;
    for(int i=1;i<=n;i++)
    {
        ans=0;
        vis[i]=1;
        dfs(i);
        if(ans)
            sum=ans/__gcd(ans,sum)*sum;
    }
    printf("%d
",sum);
    return 0;
}