小 A A A地盘上的所有人被从 1 1 1 到 n n n 编号,每个人都有自己传话的对象,第 i i i 个人对第 a i a_i ai个人传话。 有一天,小 A A A在宫殿的顶部大声喊着 O w f Owf Owf,于是一个有趣的游戏在小 A A A的地盘上开始了。
规则如下:
该游戏有许多轮,每个人都会开始一轮游戏。如果编号为 x x x 的人想要开始一轮游戏,他会对第 a x a_x ax个人说" O w w . . . w w f Oww...wwf Oww...wwf"(有 t t t 个 w w w)。如果 t > 1 t>1 t>1,第 a x a_x ax个人就会对第 a a x a_{ax} aax个人说" O w w . . . w w f Oww...wwf Oww...wwf"(有 t − 1 t−1 t−1 个 w w w)。直到有人听到" O w f Owf Owf"( t = 1 t=1 t=1),这个人就是这一轮的 J o o n Joon Joon。不存在同时进行两轮游戏的情况。 为了使游戏更有意思,小 A A A有一个邪恶的计划。他想找到最小的 t t t( t ≥ 1 t≥1 t≥1)使得对于每个人 x x x 当第 x x x 个人开始的一局游戏使 y y y 成为了 J o o n Joon Joon ,也使得由 y y y 开始的一局游戏 x x x 成为 J o o n Joon Joon 。请为小 A A A找这个最小的 t t t。 注意:可能有的人传话对象是自己。
输入格式:
第一行输入一个 n n n ( 1 ≤ n ≤ 150 1≤n≤150 1≤n≤150),表示小A地盘上的人数。
第二行输入 a 1 a_1 a1, a 2 a_2 a2, a 3 a_3 a3,… a n a_n an,第 i i i 个数表示第 i i i 个人传话的对象 a i a_i ai。
输出格式:
输出最小的 t t t,如果没有请输出 − 1 −1 −1。
样例输入:
4
2 3 1 4
样例输出:
3
解题思路:
把题中序列抽象为一张有向图,有 n n n个节点、 n n n条有向边 < i , a i > <i,a_i> <i,ai>。
如果能够达成题中所述的双向传话,两个人必须在一个环中。
如果环的长度为偶数,那么这个环的传话次数为其长度一半;
如果环的长度为奇数,那么这个环的传话次数为其长度。
我们需要做的就是统计每一个环的传话次数,然后计算最小公倍数即可。
很简单对吧qwq?
那么现在来实现代码。
首先是搜索环的长度:
void dfs(int bg) {
int next = as[bg], sum = 1;
book[bg] = true;
while (next != bg) {
if (book[next]) {
fail = true;
return;
}
sum++;
book[next] = true;
next = as[next];
}
if (sum % 2 == 0) ans.push_back(sum / 2);
else ans.push_back(sum);
}
然后计算所有ans
的最小公倍数:
long long ret = 1;
for (auto iter : ans) {
ret = lcm((long long)(iter), ret);
}
cout << ret << endl;
后排提醒:/* 十年OI一场空,不开long long见祖宗 */
。
最后,AC代码如下:
#include <iostream>
#include <vector>
using namespace std;
const int max_n = 150;
int as[max_n + 1];
bool book[max_n], fail = false;
vector<int>ans;
void dfs(int bg) {
int next = as[bg], sum = 1;
book[bg] = true;
while (next != bg) {
if (book[next]) {//出现两个人传给同一个人的情况,失败
fail = true;
return;
}
sum++;
book[next] = true;
next = as[next];
}
if (sum % 2 == 0) ans.push_back(sum / 2);
else ans.push_back(sum);
}
long long gcd(long long x, long long y) {
long long t;
while (y != 0) {
t = x % y;
x = y;
y = t;
}
return x;
}
long long lcm(long long x, long long y) {
long long ret = gcd(x, y);
return x * y / ret;
}
int main() {
int n, u, v;
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> as[i];
}
for (int i = 1; !fail && i <= n; i++) {
if (!book[i]) {
dfs(i);
}
}
if (fail) {
cout << -1 << endl;
return 0;
}
long long ret = 1;
for (auto iter : ans) {
ret = lcm((long long)(iter), ret);
}
cout << ret << endl;
return 0;
}