Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:7:35: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     long long n; scanf("%lld", &n);
                                   ^
input/code.cpp:11:54: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
                                 degin[i].insert(a[i]);
                                                      ^

#include <bits/stdc++.h>
#define MAXN 100010
using namespace std;
int a[MAXN];
set <int> degin[MAXN];
int main() {
    long long n; scanf("%lld", &n);
    long long tri = 0, broj = 0;
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]),
                                degin[a[i]].insert(i),
                                degin[i].insert(a[i]);
    for(int i = 1; i <= n; i++) {
        tri += (a[a[a[i]]] == i ? 1:0);
        broj += (long long) degin[i].size()* (n-(long long)degin[i].size()-1);
    }
    long long rez = n*(n-1)*(n-2)/6 - tri - broj/2;
    printf("%lld", rez);
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER