题目

原题链接：A. Love Triangle

题意

问所给的数字中是否有一个恰好为3的循环节。
样例：
5
2 4 5 1 3
其中s[1]->s[2]->s[4]->s[1]。

代码

#include<bits/stdc++.h>
using namespace std;
int main() {
    int n,s[5010];
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>s[i];
    }
    for(int i=1;i<=n;i++){
        if(s[s[s[s[i]]]]==s[i]){
            printf("YES\n");
            return 0;
        }
    }
    printf("NO\n");
    return 0;
}