[hihoCoder 太阁最新面经算法竞赛6] 数组重排(N个数的最小公倍数)

描述

小Hi想知道，如果他每次都按照一种固定的顺序重排数组，那么最少经过几次重排之后数组会恢复初始的顺序？

具体来讲，给定一个1 - N 的排列 P，小Hi每次重排都是把第 i 个元素放到第 Pi个位置上。例如对于 P = (2, 3, 1)，假设初始数组是(1, 2, 3)，重排一次之后变为(3, 1, 2)，重排两次之后变为(2, 3, 1)，重排三次之后变回(1, 2, 3)。

被排数组中的元素可以认为是两两不同的。

输入

第一行一个整数 N ，代表数组的长度。 (1 ≤ N ≤ 100)

第二行N个整数，代表1 - N 的一个排列 P 。

输出

输出最少重排的次数。

样例输入

3
2 3 1

样例输出

3

题解：求每个数跳到原来位置的最小跳数，循环节。再求这N个跳数的最小公倍数。注意使用long long 型！

#include<iostream>
#include<map>
#include<string>
#include<cstring>
#include<vector>
#include<algorithm>
#include<set>
#include<sstream>
#include<cstdio>
#include<unordered_map>
#include<unordered_set>
#include<cmath>
#include<climits>
using namespace std;
long long gcd(long long a, long long b)
{
   if (a < b)
      swap(a, b);
   if (b == 0)
      return a;
   else return gcd(b, a%b);
}
long long lcm(long long a, long long b)
{
    return a*b/gcd(a, b);
}
int main()
{
    int n;
    scanf("%d",&n);
    int a[101],num[101];
    for(int i=0;i<n;i++) {
        scanf("%d",&a[i]);
    }
    if(n==1) {
        printf("0\n");return 0;
    }
    for(int i=0;i<n;i++) {
        num[i]=0;
        if(a[i]-1==i) continue;
        int cur=i;
        while(true) {
            cur=a[cur]-1;
            num[i]++;
            if(cur==i) break;
        }
    }
    long long t=1;
    for(int i=0;i<n;i++) {
        if(num[i]==0) continue;
        t=lcm(t,num[i]);
    }
    cout<<t<<endl;
    return 0;
}