Witch game

#include <iostream>
#include <vector>

using namespace std;

long long n, vote[100000], c[100000];
//vector<int> v[100000];

int main() {
	cin >> n;

	for (int i = 0; i < n; i++)
	{
		cin >> vote[i];
		vote[i]--;
		if (vote[i] > i || vote[vote[i]] != i)
		{
			c[i]++;
			c[vote[i]]++;
		}
		/*if (vote[i] > i)
			v[i].push_back(vote[i]);
		else
			v[vote[i]].push_back(i);*/
	}

	long long m = n * (n - 1) * (n - 2) / 6;
	m -= n * (n - 2);
	for (int i = 0; i < n; i++)
	{
		if (vote[i] > i && vote[vote[i]] == i)
			m += (n - 2);
		if (vote[i] > i && vote[vote[i]] > i && vote[vote[vote[i]]] == i)
			m--;
		//for (int j = 0; j < v[i].size(); j++)
			//m += v[v[i][j]].size();
		//if (v[i].size() > 1)
			//m += v[i].size() * (v[i].size() - 1) / 2;
		if (c[i] > 1)
			m += c[i] * (c[i] - 1) / 2;
	}

	cout << m << endl;

	/*cout << endl << "Press enter to continue...";
	cin.ignore(numeric_limits<streamsize>::max(), '\n');
	cin.get();*/

	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