CSES - APIO 2007

Submission details
Task:	Mobiles
Sender:	Olli
Submission time:	2019-03-07 14:18:32 +0200
Language:	C++
Status:	READY
Result:	100

Feedback
group	verdict	score
#1	ACCEPTED	100

Test results
test	verdict	time
#1	ACCEPTED	0.02 s	details
#2	ACCEPTED	0.03 s	details
#3	ACCEPTED	0.02 s	details
#4	ACCEPTED	0.02 s	details
#5	ACCEPTED	0.02 s	details
#6	ACCEPTED	0.02 s	details
#7	ACCEPTED	0.02 s	details
#8	ACCEPTED	0.02 s	details
#9	ACCEPTED	0.01 s	details
#10	ACCEPTED	0.01 s	details
#11	ACCEPTED	0.01 s	details
#12	ACCEPTED	0.02 s	details
#13	ACCEPTED	0.02 s	details
#14	ACCEPTED	0.02 s	details
#15	ACCEPTED	0.01 s	details
#16	ACCEPTED	0.01 s	details
#17	ACCEPTED	0.02 s	details
#18	ACCEPTED	0.03 s	details
#19	ACCEPTED	0.01 s	details
#20	ACCEPTED	0.01 s	details
#21	ACCEPTED	0.01 s	details
#22	ACCEPTED	0.02 s	details
#23	ACCEPTED	0.01 s	details
#24	ACCEPTED	0.03 s	details
#25	ACCEPTED	0.10 s	details
#26	ACCEPTED	0.02 s	details
#27	ACCEPTED	0.01 s	details
#28	ACCEPTED	0.01 s	details
#29	ACCEPTED	0.02 s	details
#30	ACCEPTED	0.02 s	details
#31	ACCEPTED	0.03 s	details
#32	ACCEPTED	0.04 s	details
#33	ACCEPTED	0.08 s	details
#34	ACCEPTED	0.08 s	details
#35	ACCEPTED	0.08 s	details
#36	ACCEPTED	0.08 s	details
#37	ACCEPTED	0.08 s	details
#38	ACCEPTED	0.08 s	details
#39	ACCEPTED	0.08 s	details
#40	ACCEPTED	0.08 s	details
#41	ACCEPTED	0.09 s	details
#42	ACCEPTED	0.08 s	details

Code

#include <iostream>
#include <vector>
#include <set>

using namespace std;

const int N = 1e5 + 5;

int l[N];
int r[N];
int p[N];
int d[N];
int sl[N];
int sr[N];
int cl[N];
int cr[N];

int main() {
	int n;
	cin >> n;
	set<int> dep;
	for(int i = 1; i <= n; ++i) {
		int a, b;
		cin >> a >> b;
		l[i] = a;
		r[i] = b;
		if(a != -1) {
			p[a] = i;
			d[a] = d[i] + 1;
		} else {
			dep.insert(d[i] + 1);
		}
		if(b != -1) {
			p[b] = i;
			d[b] = d[i] + 1;
		} else {
			dep.insert(d[i] + 1);
		}
	}

	if(dep.size() > 2) {
		cout << -1 << "\n";
		return 0;
	} else if (dep.size() == 1) {
		cout << 0 << "\n";
		return 0;
	} else {
		auto it = dep.begin();
		int a = *it;
		++it;
		int b = *it;
		if(b - a != 1) {
			cout << -1 << "\n";
			return 0;
		}
	}


	int hi = *(dep.begin());

	for(int i = n; i >= 1; --i) {
		if(l[i] != -1) {
			sl[i] = sl[l[i]] + sr[l[i]];
		} else {
			sl[i] = 1;
		}
		if(r[i] != -1) {
			sr[i] = sl[r[i]] + sr[r[i]];
		} else {
			sr[i] = 1;
		}

		if(d[i] == hi - 1) {
			if(l[i] == -1) {
				cl[i] = 1;
			}
			if(r[i] == -1) {
				cr[i] = 1;
			}
		} else {
			if(l[i] != -1) {
				cl[i] = cl[l[i]] + cr[l[i]];
			} else {
				cl[i] = 0;
			}
			if(r[i] != -1) {
				cr[i] = cr[r[i]] + cl[r[i]];
			} else {
				cr[i] = 0;
			}
		}
	}




	int ans = 0;
	int cu = 1;

	while(cu != -1) {
		int i = cu;
		if(cl[i] == 0 && cr[i] == 0) break;
		if(cl[i] > 0 && cl[i] < sl[i] && cr[i] > 0 && cr[i] < sr[i]) {
			cout << -1 << "\n";
			return 0;
		}
		if(cr[i] == sr[i]) {
			cu = l[cu];
			continue;
		}
		if(cr[i] == 0) {
			++ans;
			cu = l[cu];
			continue;
		}
		//0 < cr[i] < sr[i]

		if(cl[i] != sl[i] && cl[i] != 0) {
			cout << -1 << "\n";
			return 0;
		}

		if(cl[i] == 0) {
			cu = r[cu];
			continue;
		}
		++ans;
		cu = r[cu];
	}

	cout << ans << "\n";


}

Test details

Test 1

Verdict: ACCEPTED

input
`1 -1 -1` view save

correct output
`0` view save

user output
`0` view save

Test 2

Verdict: ACCEPTED

input
`2 2 -1 -1 -1` view save

correct output
`0` view save

user output
`0` view save

Test 3

Verdict: ACCEPTED

input
`2 -1 2 -1 -1` view save

correct output
`1` view save

user output
`1` view save

Test 4

Verdict: ACCEPTED

input
`4 2 3 4 -1 -1 -1 -1 -1` view save

correct output
`0` view save

user output
`0` view save

Test 5

Verdict: ACCEPTED

input
`4 2 3 -1 4 -1 -1 -1 -1` view save

correct output
`1` view save

user output
`1` view save

Test 6

Verdict: ACCEPTED

input
`4 2 3 -1 -1 4 -1 -1 -1` view save

correct output
`1` view save

user output
`1` view save

Test 7

Verdict: ACCEPTED

input
`4 2 3 -1 -1 -1 4 -1 -1` view save

correct output
`2` view save

user output
`2` view save

Test 8

Verdict: ACCEPTED

input
`5 2 3 4 5 -1 -1 -1 -1 ...` view save

correct output
`0` view save

user output
`0` view save

Test 9

Verdict: ACCEPTED

input
`5 2 3 4 -1 5 -1 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 10

Verdict: ACCEPTED

input
`5 2 3 4 -1 -1 5 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 11

Verdict: ACCEPTED

input
`5 2 3 -1 4 5 -1 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 12

Verdict: ACCEPTED

input
`5 2 3 -1 4 -1 5 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 13

Verdict: ACCEPTED

input
`5 2 3 -1 -1 4 5 -1 -1 ...` view save

correct output
`1` view save

user output
`1` view save

Test 14

Verdict: ACCEPTED

input
`6 2 3 4 5 6 -1 -1 -1 ...` view save

correct output
`0` view save

user output
`0` view save

Test 15

Verdict: ACCEPTED

input
`6 2 3 4 5 -1 6 -1 -1 ...` view save

correct output
`1` view save

user output
`1` view save

Test 16

Verdict: ACCEPTED

input
`6 2 3 4 -1 5 6 -1 -1 ...` view save

correct output
`1` view save

user output
`1` view save

Test 17

Verdict: ACCEPTED

input
`6 2 3 -1 4 5 6 -1 -1 ...` view save

correct output
`2` view save

user output
`2` view save

Test 18

Verdict: ACCEPTED

input
`19 2 3 6 5 4 7 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 19

Verdict: ACCEPTED

input
`9 3 2 4 5 -1 -1 6 9 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 20

Verdict: ACCEPTED

input
`4 -1 2 3 4 -1 -1 -1 -1` view save

correct output
`-1` view save

user output
`-1` view save

Test 21

Verdict: ACCEPTED

input
`12 3 2 4 5 -1 6 7 9 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 22

Verdict: ACCEPTED

input
`10 3 2 7 5 4 6 8 10 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 23

Verdict: ACCEPTED

input
`1000 2 -1 -1 3 4 -1 5 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 24

Verdict: ACCEPTED

input
`10000 2 -1 -1 3 4 -1 5 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 25

Verdict: ACCEPTED

input
`100000 2 -1 -1 3 4 -1 5 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 26

Verdict: ACCEPTED

input
`10 2 3 6 5 7 4 -1 -1 ...` view save

correct output
`2` view save

user output
`2` view save

Test 27

Verdict: ACCEPTED

input
`18 2 3 7 6 5 4 11 9 ...` view save

correct output
`3` view save

user output
`3` view save

Test 28

Verdict: ACCEPTED

input
`13 3 2 4 5 7 6 -1 -1 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 29

Verdict: ACCEPTED

input
`660 3 2 5 6 7 4 10 8 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 30

Verdict: ACCEPTED

input
`1250 2 3 6 7 5 4 10 14 ...` view save

correct output
`7` view save

user output
`7` view save

Test 31

Verdict: ACCEPTED

input
`5000 2 3 6 7 5 4 11 14 ...` view save

correct output
`4` view save

user output
`4` view save

Test 32

Verdict: ACCEPTED

input
`32767 2 3 5 4 7 6 13 14 ...` view save

correct output
`0` view save

user output
`0` view save

Test 33

Verdict: ACCEPTED

input
`100000 2 3 5 7 4 6 15 14 ...` view save

correct output
`7` view save

user output
`7` view save

Test 34

Verdict: ACCEPTED

input
`98348 3 2 7 5 6 4 10 8 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 35

Verdict: ACCEPTED

input
`100000 2 3 5 7 4 6 15 14 ...` view save

correct output
`7` view save

user output
`7` view save

Test 36

Verdict: ACCEPTED

input
`99999 3 2 7 5 6 4 10 8 ...` view save

correct output
`5` view save

user output
`5` view save

Test 37

Verdict: ACCEPTED

input
`98348 3 2 7 5 6 4 10 8 ...` view save

correct output
`-1` view save

user output
`-1` view save

Test 38

Verdict: ACCEPTED

input
`98303 3 2 5 4 7 6 9 8 ...` view save

correct output
`1` view save

user output
`1` view save

Test 39

Verdict: ACCEPTED

input
`98304 3 2 5 4 7 6 9 8 ...` view save

correct output
`16` view save

user output
`16` view save

Test 40

Verdict: ACCEPTED

input
`99989 3 2 5 4 7 6 9 8 ...` view save

correct output
`15` view save

user output
`15` view save

Test 41

Verdict: ACCEPTED

input
`99989 2 3 4 5 6 7 8 9 ...` view save

correct output
`15` view save

user output
`15` view save

Test 42

Verdict: ACCEPTED

input
`100000 3 2 5 4 7 6 9 8 ...` view save

correct output
`16` view save

user output
`16` view save

APIO 2007

Mobiles

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Test 32

Test 33

Test 34

Test 35

Test 36

Test 37

Test 38

Test 39

Test 40

Test 41

Test 42

Tasks

Submissions (Olli)