CodeForces -939A 三角恋爱

题目链接：
http://codeforces.com/problemset/problem/939/A
A. Любовный треугольник
ограничение по времени на тест1 секунда
ограничение по памяти на тест256 мегабайт
вводстандартный ввод
выводстандартный вывод
Как известно, нет самолетов-мужчин и самолетов-женщин. Однако, каждому самолету на Земле нравится какой-то один другой самолет. Всего на Земле n самолетов, пронумерованных от 1 до n, при этом самолету номер i нравится самолет fi, где 1 ≤ fi ≤ n, а также fi ≠ i.

Назовем любовным треугольником ситуацию, когда самолету A нравится самолет B, самолету B нравится самолет C, а самолету C нравится самолет A. Проверьте, есть ли на Земле хотя бы один любовный треугольник.

Входные данные
Первая строка содержит одно целое число n (2 ≤ n ≤ 5000) — число самолетов.

Вторая строка содержит n целых чисел f1, f2, …, fn (1 ≤ fi ≤ n, fi ≠ i), что означает, что i-му самолету нравится fi-й.

Выходные данные
Выведите «YES», если на Земле есть любовный треугольник из самолетов. Иначе выведите «NO».

Каждую букву можно выводить в любом регистре: строчную или заглавную.

Примеры
входные данныеСкопировать
5
2 4 5 1 3
выходные данныеСкопировать
YES
входные данныеСкопировать
5
5 5 5 5 1
выходные данныеСкопировать
NO
Примечание
В первом примере самолету 2 нравится самолет 4, самолету 4 нравится самолет 1, а самолету 1 нравится самолет 2, поэтому это любовный треугольник.

Во втором примере любовных треугольников нет.

A.三角恋爱

Time limit1000 ms	Memory limit262144 kB

在幼儿园中有 n 个小朋友, 这些小朋友从 1 到 n 标号，每个小朋友都有一个特别想要一起玩的同伴，标号为 i 的小朋友特别希望能够和标号为 fi 的小朋友一起玩, 其中 1 ≤ fi ≤ n 且 fi ≠ i.

如果存在三个小朋友 A, B, C 满足 A 想和 B 玩, B 想和 C 玩，C 想和 A 玩，那么这三个小朋友就被称为一个三人游戏组，给出每个小朋友最想和哪个小朋友一起玩，你需要判断其中是否存在一个三人游戏组

Input
输入数据的第一行包含一个正整数 n (2 ≤ n ≤ 5000) — 表示小朋友的人数.

第二行 n 个整数 f1, f2, …, fn (1 ≤ fi ≤ n, fi ≠ i), 表示第 i 个小朋友最想和第 fi个小朋友玩.

Output
如果能够找到一个游戏三人组，输出“YES”， 否则输出“NO”。（引号不需要输出）

Samlpe Input1
5
2 4 5 1 3
Sample Output1
YES
Samlpe Input2
5
5 5 5 5 1
Sample Output2
NO

Note
在第一个样例中，2 想要和 4 玩, 4 想要和 1 玩, 1 想要和2 玩， 这三个小朋友就构成了一个游戏三人组.
在第二组样例中不存在满足条件的这样三个小朋友.

套三个数组就行，看看有没有值

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;

int main()
{
    int n, a[5005];
    while(scanf("%d", &n)!=EOF)
    {
        for(int i=1; i<=n; i++)
            scanf("%d", &a[i]);
        int flag=0;
        for(int i=1; i<=n; i++)
        {
            if(i==a[a[a[i]]])
            {
                flag=1;
                break;
            }
        }
        if(flag)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}