week1-[一维数组]传送

week1-[一维数组]传送

题目描述

有 $n$ 个传送门，从第 $i$ 个传送门进去后会被传送到第 $a_i$ 个传送门，进而被传送到第 $a_{a_i}$ 个传送门，如此一直下去……小 A 想知道从第 $k$ 个传送门进去后，能不能回到第 $k$ 个传送门？

输入格式

输入共 $2$ 行。

第 $1$ 行输入 $2$ 个正整数 $n,k$。

第 $2$ 行输入 $n$ 个正整数 $a_i$。

输出格式

输出共 $1$ 行表示答案。如果可以，输出 GO BACK 2 YOUR RAVE，否则输出 CHAIN1761BPM170。

样例 #1

样例输入 #1

5 3
1 3 4 5 2

样例输出 #1

GO BACK 2 YOUR RAVE

提示

样例解释 $1$

从第 $3$ 个传送门进去后到了第 $a_3=4$ 个传送门。

从第 $4$ 个传送门进去后到了第 $a_4=5$ 个传送门。

从第 $5$ 个传送门进去后到了第 $a_5=2$ 个传送门。

从第 $2$ 个传送门进去后到了第 $a_2=3$ 个传送门，成功回来了。

数据范围

对于所有数据，$1\le n\le 100$，$1\le a_i\le n$。

📄 C++ 代码实现：

#include <iostream>
using namespace std;int main() {int n, k;cin >> n >> k;int a[101]; // 下标从1到nfor(int i = 1; i <= n; i++) {cin >> a[i];}bool visited[101] = {false};int current = k;while(true) {if(current == k && visited[k]) { // 回到起点且至少走过一次cout << "GO BACK 2 YOUR RAVE" << endl;break;}if(visited[current]) { // 进入了其他循环cout << "CHAIN1761BPM170" << endl;break;}visited[current] = true;current = a[current];}return 0;
}