CSES - Datatähti Open 2017 - Results
Submission details
Task:Witch game
Sender:Lumibons
Submission time:2017-01-22 19:24:50 +0200
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.03 s1details
#2ACCEPTED0.04 s1details
#30.04 s1details
#4ACCEPTED0.04 s1details
#5ACCEPTED0.05 s1details
#60.04 s2details
#7ACCEPTED0.05 s2details
#80.04 s2details
#9ACCEPTED0.05 s2details
#10ACCEPTED0.04 s2details
#110.07 s3details
#12ACCEPTED0.07 s3details
#130.07 s3details
#14ACCEPTED0.09 s3details
#150.09 s3details

Code

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

Test 1

Group: 1

Verdict:

input
100
2 1 4 3 6 5 8 7 10 9 12 11 14 ...

correct output
156800

user output
156900

Test 2

Group: 1

Verdict: ACCEPTED

input
100
2 3 4 5 6 7 8 9 10 11 12 13 14...

correct output
152000

user output
152000

Test 3

Group: 1

Verdict:

input
100
2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
156849

user output
156949

Test 4

Group: 1

Verdict: ACCEPTED

input
100
2 3 1 5 6 4 8 9 7 11 12 10 14 ...

correct output
151968

user output
151968

Test 5

Group: 1

Verdict: ACCEPTED

input
100
8 98 100 62 42 36 95 70 22 49 ...

correct output
152040

user output
152040

Test 6

Group: 2

Verdict:

input
5000
2 1 4 3 6 5 8 7 10 9 12 11 14 ...

correct output
20808340000

user output
20808345000

Test 7

Group: 2

Verdict: ACCEPTED

input
5000
2 3 4 5 6 7 8 9 10 11 12 13 14...

correct output
20795850000

user output
20795850000

Test 8

Group: 2

Verdict:

input
5000
2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
20808342499

user output
20808347499

Test 9

Group: 2

Verdict: ACCEPTED

input
5000
2 3 1 5 6 4 8 9 7 11 12 10 14 ...

correct output
20795848337

user output
20795848337

Test 10

Group: 2

Verdict: ACCEPTED

input
5000
283 2880 2565 3289 4160 936 39...

correct output
20795852465

user output
20795852465

Test 11

Group: 3

Verdict:

input
100000
2 1 4 3 6 5 8 7 10 9 12 11 14 ...

correct output
166656666800000

user output
166656666900000

Test 12

Group: 3

Verdict: ACCEPTED

input
100000
2 3 4 5 6 7 8 9 10 11 12 13 14...

correct output
166651667000000

user output
166651667000000

Test 13

Group: 3

Verdict:

input
100000
2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
166656666849999

user output
166656666949999

Test 14

Group: 3

Verdict: ACCEPTED

input
100000
2 3 1 5 6 4 8 9 7 11 12 10 14 ...

correct output
166651666966668

user output
166651666966668

Test 15

Group: 3

Verdict:

input
100000
186 62491 95379 37431 88427 93...

correct output
166651667250100

user output
166651667250108