AcWing 4626. 最小移动距离

链接：4626. 最小移动距离 - AcWing题库

平面上有 n 个点，编号为 1∼n。

对于每个点 i（1≤i≤n），都存在一条从点 i 到点 ai（1≤ai≤n，ai 可以等于 i）的有向边。

所有边的长度均为 1。

请你判断是否存在一个最小移动距离 t（t≥1），使得：

我们规定，如果从点 u 出发，移动 t 单位长度距离后，到达点 v，就称点 v 是点 u 的目标点。注意，一个点的目标点也可能是它自己。
对于图中的每个点 x，如果点 y 是点 x 的目标点，则点 x 也必须是点 y 的目标点。
如果存在这样的 t，请你输出 t 的最小可能值，否则请你输出 -1。

输入格式
第一行包含一个整数 n。

第二行包含 n 个整数 a1,a2,…,an。

输出格式
如果存在满足条件的 t（t≥1），则输出一个正整数，表示 t 的最小可能值。

否则输出 -1。

数据范围
前 3 个测试点满足 1≤n≤4。
所有测试点满足 1≤n≤100，1≤ai≤n。

输入样例1：

4
2 3 1 4

输出样例1：

3

输入样例2：

4
4 4 4 4

输出样例2：

-1

输入样例3：

4
2 1 4 3

输出样例3：

1

由于题目要求所有的点ai(i=1,2,..,n)，都存在ai移动到aj(j=1,2,..,n)的移动次数t，也能够让aj移动到ai。所以ai到aj之间必须存在回路。 如果不存在回路就直接输出-1.

当某回路长度为奇数时，在该回路上的点ai满足题目条件(即ai->aj的移动次数等于aj->ai的移动次数)的最小移动次数就等于回路长度(当回路长度为奇数时，要让ai->aj的移动次数等于aj->ai，只有可能ai==aj)。

当某回路长度为偶数时，在该在回路上的点ai满足题目条件的最小移动次数，则等于回路长度除以2。

所以要求出最小的t,让图上所有的点ai满足题目条件，只要求所有回路的最小移动次数的最小公倍数即可。  

代码如下：

#include<iostream>
#include<algorithm>
#include<cstring> 
#include<vector>
using namespace std;
typedef long long LL; 
int e[105]; //e[i]用来存储i指向的下一个点 
int fg[105] {0}; //标记用数组，用来标记 
vector<int> raw; //用来存储每条回路的最小移动次数 
LL GetGcd(LL x, LL y) //求最大公约数 
{
	if(x<y) swap(x, y);
	while(x%y)
	{
		int t = x % y;
		x = y;
		y = t;
	}
	return y;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n;
	cin >> n;
	memset(e, -1, sizeof(-1)); //初始话e数组 
	for(int i=1; i<=n; i++)
	{
		int x;
		cin >> x;
		e[i] = x;	
	}
	LL maxv = 0; //答案 
	for(int i=1; i<=n; i++) //遍历每一个点，从该点出发，判断该点是否在回路上。 
	{
		int v = i, cnt = 0; //cnt记录移动次数 
		memset(fg, 0, sizeof(fg)); //初始化标记数组 
		while(1)
		{
			v = e[v];
			if(v==-1 || fg[v] && v!=i) //当该点不存在或该点被标记但不是起点，则表示起点不在回路上，直接退出。 
			{
				cnt = -1;
				break;
			}
			fg[v] ++; //标记经过的点 
			cnt ++; //移动次数加一 
			if(v==i) break; //当该点等于起点时，表示已经走完一个回路，退出 
		}
		if(cnt>0) raw.push_back((cnt&1? cnt:cnt/2)); //如果存在回路，记录每个回路的最小移动次数 
		else //有点不在回路上，maxv = -1 
		{
			maxv = -1;
			break;
		}
	}
	if(!maxv) //当maxv=0时表示结果存在 
	{
		maxv = raw[0];
		for(int i=1; i<raw.size(); i++) //求所有回路最小移动次数的最小公倍数 
			maxv = maxv*raw[i] / GetGcd(maxv, raw[i]);
	}
	cout  << maxv<< endl;
	return 0;
}