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:26:17: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",&p);
                 ^

#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];

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++){
		int p;
		scanf("%d",&p);
		gph[i].push_back(p);
		gph[p].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;
	int chk[100005];
	for(int i=1; i<=n; i++){
		ans += nc2(n - gph[i].size() - 1);
		memset(chk, 0, sizeof(chk));
		ans -= n;
		int cyc3 = 0;
		for(auto &j : gph[i]){
			if(chk[j]) cyc3++;
			chk[j] = 1;
			ans += gph[j].size();
		}
		if(cyc3) ans-=2;
	}
	cout << ans/3;
}

Test 1

Group: 1

Verdict: WRONG ANSWER

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

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