제출 #1366130

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1366130		sofapuden	Cat in a tree (BOI17_catinatree)	C++20	100 / 100	21 ms	3560 KiB

catinatree

#include<bits/stdc++.h>

using namespace std;

int main() {
	ios_base::sync_with_stdio(0); cin.tie(0);
	int n, d; cin >> n >> d;
	vector<int> dep(n), P(n);
	for(int i = 1; i < n; ++i) {
		cin >> P[i];
		dep[i] = dep[P[i]] + 1;
	}

	vector<int> od(n);
	iota(od.begin(), od.end(), 0);
	sort(od.begin(), od.end(), [&](auto a, auto b) { return dep[a] > dep[b]; });

	vector<int> dis(n, d);
	int ans = 0;

	for(auto i : od) {
		int c = i;
		bool ok = 1;
		for(int i = 0; i < d && ok; ++i) {
			if(dis[c] + i < d) ok = 0;
			dis[c] = i;
			if(!c) break;
			c = P[c];
		}
		ans += ok;
	}
	cout << ans << '\n';
}

• 서브태스크 1: 11 / 11
• 서브태스크 2: 40 / 40
• 서브태스크 3: 49 / 49