Witch game

#include <iostream>

using namespace std;

int n, x[100000], ingoing[100000];

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) {
        cin >> x[i];
        x[i]--;
        ingoing[x[i]]++;
    }

    long long res = 0;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            if (i == j || x[i] == j || x[j] == i) continue;
            int cur = n - 4 - ingoing[i] - ingoing[j];
            if (x[i] == x[j]) cur++;
            if (x[x[i]] == i || x[x[i]] == j) cur++;
            if (x[j] != x[i] && (x[x[j]] == i || x[x[j]] == j)) cur++;
            res += (long long) cur;
            //cout << i << " " << j << " " << cur << "\n";
        }
    }
    cout << res / 6 << "\n";
}

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