Witch game

input/code.cpp: In function 'int main()':
input/code.cpp:23:16: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d",&n);
                ^
input/code.cpp:25:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",&p[i]);
                    ^

#include <bits/stdc++.h>
using namespace std;
typedef long long lint;
typedef long double llf;

int n;
vector<int> gph[100005];
vector<int> dfn;
int vis[100005], sz[100005], p[100005];

void dfs(int x){
	if(vis[x]) return;
	vis[x] = 1;
	dfn.push_back(x);
	for(auto &i : gph[x]) dfs(i);
}

lint nc2(lint x){
	return x * (x-1) / 2;
}

int main(){
	scanf("%d",&n);
	for(int i=1; i<=n; i++){
		scanf("%d",&p[i]);
		gph[i].push_back(p[i]);
		gph[p[i]].push_back(i);
	}
	for(int i=1; i<=n; i++){
		if(!vis[i]){
			dfn.clear();
			dfs(i);
			for(auto &j : dfn){
				sz[j] = dfn.size();
			}
		}
	}
	lint ans = 0;
	for(int i=1; i<=n; i++){
		if(p[p[p[i]]] == i) ans++;
		ans += nc2(n - gph[i].size() - 1);
		for(auto &j : gph[i]){
			ans += gph[j].size();
		}
		ans -= n;
	}
	cout << ans/3;
}

Test 1

Group: 1

Verdict: WRONG ANSWER

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: WRONG ANSWER

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: WRONG ANSWER

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: WRONG ANSWER

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER