Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:8:18: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%lli",&N);
                  ^
input/code.cpp:19:25: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lli",&data[i]);
                         ^

#include <bits/stdc++.h>
#define MAXN 100005

using namespace std;

int main () {
	long long N;
	scanf("%lli",&N);
	if (N == 3LL) {
		printf("0\n");
		return 0;
	}
	static long long cnt[MAXN];
	static long long data[MAXN];
	memset(cnt, 0, sizeof(cnt));
	long long i;
	long long M = 0LL;
	for (i = 0; i < N; i++) {
		scanf("%lli",&data[i]);
		data[i]--;
		cnt[i]++;
		cnt[data[i]]++;
		M++;
	}
	for (i = 0; i < N; i++) {
		if (i < data[i]) {
			if (data[data[i]] == i) {
				cnt[i]--;
				cnt[data[i]]--;
				M--;
			}
		}
	}
	long long res = 0LL;
	res += N * (N - 1LL) * (N - 2LL);
	res -= 6LL * M * (N - 2LL);
	for (i = 0; i < N; i++) {
		res += 3LL * cnt[i] * (cnt[i] - 1LL);
		if (data[data[data[i]]] == i) res -= 2LL;
	}
	res /= 6LL;
	printf("%lli\n",res);
	return 0;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

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: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED