Witch game

#include <bits/stdc++.h>
#define SZ(x) ((int) (x).size())
using namespace std;

typedef long long int64;

const int NMAX = 5000;

int frequency[NMAX];
int value[NMAX];

int main(void) {
    cin.tie(0);
    ios_base::sync_with_stdio(false);
    
    int n; cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> value[i];
        value[i]--;
        frequency[value[i]]++;
    }
    
    int64_t answer = 0LL;
    for (int i = 0; i < n; i++) {
        memset(frequency, 0, 4 * n);
        for (int j = i + 1; j < n; j++) {
            frequency[value[j]]++;
        }
        for (int j = i + 1; j < n - 1; j++) {
            frequency[value[j]]--; 
            if (value[i] == j || value[j] == i) {
                continue;
            }
            answer += (n - j - 1) - frequency[i] - frequency[j];
            if (value[i] == value[j]) {
                if (value[i] > j && (value[value[i]] != i && value[value[i]] != j)) {
                    answer--;
                }
            } else {
                if (value[i] > j && (value[value[i]] != i && value[value[i]] != j)) {
                    answer--;
                }
                if (value[j] > j && (value[value[j]] != i && value[value[j]] != j)) {
                    answer--;
                }
            }
        }
    }
    cout << answer << '\n';
    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: RUNTIME ERROR

Test 12

Group: 3

Verdict: RUNTIME ERROR

Test 13

Group: 3

Verdict: RUNTIME ERROR

Test 14

Group: 3

Verdict: RUNTIME ERROR

Test 15

Group: 3

Verdict: RUNTIME ERROR