제출 #884826

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
884826		42kangaroo	참나무 (IOI23_beechtree)	C++17	17 / 100	2084 ms	152468 KiB

beechtree

//  D. Beech Tree

#include<bits/stdc++.h>
#include "beechtree.h"

using namespace std;

using g_t = vector<vector<int>>;
using dVal = tuple<map<int, int>, list<set<pair<int,int>>>, int>;

struct ValR {
	int d, n, co;
};

dVal dfs(int n, g_t &g, vector<bool> &po, vector<int> &c) {
	map<int, int> r;
	map<int, vector<pair<int, int>>> ve;
	list<set<pair<int,int>>> se;
	int seSi = 0;
	for (auto e: g[n]) {
		if (r.find(c[e]) != r.end()) po[n] = false;
		r[c[e]] = 1;
		ve[c[e]] = {};
	}
	for (auto e: g[n]) {
		auto [re, li, si] = dfs(e, g, po, c);
		if (!po[e]) {
			po[n] = false;
		}
		if (po[n]) {
			for (auto &[co, d]: re) {
				if (r.find(co) == r.end()) {
					po[n] = false;
					break;
				}
				if (co == c[e]) r[c[e]] = d + 1;
				ve[co].emplace_back(d, e);
			}
			if (po[n]) {
				if (si > seSi) {
					se.swap(li);
					swap(si, seSi);
				}
				auto itS = se.begin();
				for (auto & it : li) {
					while (itS != se.end() && itS->size() <= it.size()) {
						for (auto cos: *itS) {
							if (it.erase(cos) == 0) {
								po[n] = false;
								break;
							}
						}
						if (!po[n]) break;
						itS++;
					}
					if (po[n]) {
						if (!it.empty()) {
							auto act = itS;
							itS = se.insert(itS, set<pair<int,int>>());
							for (auto cos: it) {
								if (act->erase(cos) == 0) {
									po[n] = false;
									break;
								}
								itS->insert(cos);
							}
						}
					}
				}
			}
		}
	}
	if (!po[n]) return {};
	map<int, int> val;
	se.emplace_back();
	for (auto &[co, vec]: ve) {
		bool dD = true;
		for (auto [d, e]: vec) {
			val[e] = 0;
			if (d > r[co]) {
				po[n] = false;
				return {};
			} else if (d == r[co]) {
				dD = false;
				break;
			}
		}
		if (dD) {
			se.back().insert({co, r[co]});
			seSi++;
		}
	}
	if (se.back().empty()) se.pop_back();
	return {r, se, seSi};
}

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C) {
	g_t g(N);
	for (int i = 1; i < N; ++i) {
		g[P[i]].push_back(i);
	}
	vector<bool> po(N, true);
	dfs(0, g, po, C);
	return {po.begin(), po.end()};
}

• Subtask 0: 0 / 0
• Subtask 1: 9 / 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