Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:14:15: warning: unused variable 'x' [-Wunused-variable]
     long long x=1000000;
               ^

#include <iostream>
#include <assert.h>
#include <map>
using namespace std;
long long a[100002],j[100002];
int main(){
    long long ans,n;
    cin>>n;
    if(n==3){
        assert(0);
        return 0;
    }
    ans=n*(n-1)*(n-2)/6;
    long long x=1000000;
    for(long long i=1;i<=n;i++){
        cin>>j[i];
        if(j[i]>i || j[j[i]]!=i){
            ans-=n-2-a[i]-a[j[i]];
            if(j[j[j[i]]]==i)
                ans--;
            a[i]++;
            a[j[i]]++;
        }
    }
    cout<<ans<<endl;
    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