Codeforces 524C Idempotent functions

题目链接：http://codeforces.com/problemset/problem/542/C

题目大意：给定一种运算f，对于输入的数组来说，一步操作，f(x) = a[x]，两步操作，f^2(x) = a[a[x]]....倘若每次进行k步操作之后，得到的都是同一个数，即k为符合条件的步数。求出能令所有数都符合条件的最小步数。

初步我们可以很快想到，如果能求出每个数的符合条件的最小步数，然后求出这些步数的最小公倍数即可。但是需要注意的是，对于一开始并没有直接进入循环的情况，即"6"字形的情况，我们要分开求出多余的长度len[i]和循环节长度step[i]，选出最长的MAXlen，ans即为MAXlen对所有step[i]的最小公倍数的上取整。

AC代码：

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#include <cmath>
#include <algorithm>
using namespace std;
#define MAXN 500
typedef long long LL;
int n;
int a[MAXN];
LL step[MAXN];
int vis[MAXN], v[MAXN];
int main() {
    LL ans, maxNum = 0;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    for(int i = 1; i <= n; i++) {
        memset(vis, 0, sizeof(vis));
        stack<int> s;
        int j;
        for(j = i; !vis[j]; j = a[j]) {
            vis[j] = 1;
            s.push(j);
        }
        LL num = 1;
        while(s.top() != j) {
            s.pop(); num++;
        }
        s.pop();
        step[i] = num;
        maxNum = max(maxNum, (LL)s.size());
    }
    ans = step[1];
    for(int i = 1; i <= n; i++) {
        ans = ans / __gcd(ans, step[i]) * step[i];
    }
    LL temp = maxNum / ans;
    if(maxNum % ans || maxNum == 0) temp++;
    ans = temp * ans;
    printf("%I64d\n", ans);
    return 0;
}