Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:13:21: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
    scanf("%lld", &n);
                     ^
input/code.cpp:15:27: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
       scanf("%d", &out[i]);
                           ^

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

typedef long long Long;

const int MAXN = 1e5;

Long n;
int out[MAXN + 10];
Long in[MAXN + 10];

int main() {
   scanf("%lld", &n);
   for(int i = 1; i <= n; i++) {
      scanf("%d", &out[i]);
      in[out[i]]++;
   }
   Long cc = n * (n - 1) * (n - 2) / 6;
   Long bad = 0;
   for(int u = 1; u <= n; u++) {
      int v = out[u];
      if(out[v] == u) {
         if(u < v) bad += n-2;
         bad += (in[u]-1)*(in[u]-2) / 2;
      } else {
         bad += (in[u] * (in[u] - 1)) / 2;
         if(out[out[v]] == v) {
            continue;
         }
         bad += n - in[u] - in[v] - 1;
      }
   }
   // cerr << "bad = " << bad << "\n";
   cc -= bad;
   printf("%lld\n", cc);
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER