Submission #1071385

#	Time	Username	Problem	Language	Result	Execution time	Memory
1071385		UmairAhmadMirza	Beech Tree (IOI23_beechtree)	C++17	8 / 100	103 ms	60496 KiB

beechtree

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

int const N=2e5+5;
vector<int> child[N];
set<int> ch_col[N];
vector<int> ans;
int dep[N];
void dfs(int node){
	ans[node]=1;
	vector<int> ch;
	for(int i:child[node]){
		dfs(i);
		dep[node]=max(dep[i]+1,dep[node]);
		if(dep[i]==1)
			ch.push_back(i);
	}
	if(ch.size()>1 || dep[node]>2 || child[node].size()!=ch_col[node].size()){
		ans[node]=0;
		return;
	}
	if(ch.size()==0)
		return;
	if(ans[ch[0]]==0)
		ans[node]=0;
	for(int c:ch_col[ch[0]])
		if(ch_col[node].find(c)==ch_col[node].end())
			ans[node]=0;
	return;
}
vector<int> beechtree(int n, int M, vector<int> P, vector<int> C){
	for (int i = 1; i < n; ++i){
		child[P[i]].push_back(i);
		ch_col[P[i]].insert(C[i]);
	}
	ans.resize(n);
	for (int i = 0; i < n; ++i)
		ans[i]=1;
	dfs(0);
	return ans;
}

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