Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:17:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
                    ^
input/code.cpp:18:52: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
                                                    ^

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef pair<int, int> II;

const int N = (int) 1e5 + 10;
int n, a[N];
bool mark[N];

int main() {
    #ifdef LOCAL
        freopen("Data.inp", "r", stdin);
        freopen("Data.out", "w", stdout);
    #endif

    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);

    LL ans = 0;
    for (int z = 1; z <= n; ++z) {
        int c1 = 0, c2 = 0;
        for (int i = 1; i <= n; ++i) {
            mark[i] = a[z] != i && a[i] != z;
            c1 += mark[i] && i < z;
            c2 += mark[i] && i > z;
        }
        LL sum = c1 * c2;
        for (int i = 1; i <= z - 1; ++i)
            if (mark[i] && mark[a[i]] && a[i] > z) sum--;
        for (int i = z + 1; i <= n; ++i)
            if (mark[i] && mark[a[i]] && a[i] < z) sum--;
        for (int i = 1; i <= z - 1; ++i)
            if (mark[i] && mark[a[i]] && a[i] > z && a[a[i]] == i) sum++;
        ans += sum;
    }

    cout << ans;
    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: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED