Submission #1046491

#	Time	Username	Problem	Language	Result	Execution time	Memory
1046491		0npata	Beech Tree (IOI23_beechtree)	C++17	9 / 100	53 ms	23132 KiB

beechtree

#include "beechtree.h"
#include<bits/stdc++.h>
using namespace std;

#define vec vector

std::vector<int> beechtree(int N, int M, std::vector<int> P, std::vector<int> C)
{
	vec<set<int>> cs(N);
	vec<int> ans(N, 1);

	bool ok = true;
	for(int i = 1; i<N; i++) {
		if(cs[P[i]].count(C[i])) {
			ok = false;
			ans[P[i]] = 0;
		}
		cs[P[i]].insert(C[i]);
	}

	ans[0] = ok;

	if(!ok) return ans;

	sort(cs.begin()+1, cs.end(), [&](auto& s1, auto& s2) { return s1.size()>s2.size(); });

	for(int i = 1; i<N; i++) {
		for(int val : cs[i]) {
			if(!cs[i-1].count(val)) {
				ans[0] = false;
				return ans;
			}
		}
	}

	return ans;

}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 9
• Subtask 2: 0 / 5
• Subtask 3: 9 / 9
• Subtask 4: 0 / 8
• Subtask 5: 0 / 14
• Subtask 6: 0 / 14
• Subtask 7: 0 / 12
• Subtask 8: 0 / 17
• Subtask 9: 0 / 12