이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int n, d, par[N], dist[N], vis[N];
bool mark[N];
vector<int> g[N];
void dfs(int v, int p = -1){
for (int u : g[v]){
if (u == p) continue;
dist[u] = dist[v] + 1;
dfs(u, v);
}
}
void dfs_vis(int v, int start, int cur = 0, int p = -1){
if (cur + 1 == d) return;
for (int u : g[v]){
if (u == p) continue;
int my = (d - vis[u] + 1) * (bool(vis[u]));
if (my > (d - vis[v])) continue;
vis[u] = vis[v] + 1;
dfs_vis(u, start, cur + 1, v);
}
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
cin >> n >> d;
for (int i = 1; i < n; i ++){
cin >> par[i];
g[par[i]].push_back(i);
g[i].push_back(par[i]);
}
dfs(0);
vector<pair<int, int>> vec;
for (int i = 0; i < n; i ++)
vec.push_back({dist[i], i});
sort(vec.begin(), vec.end());
int ans = 0;
while (!vec.empty()){
auto p = vec.back();
vec.pop_back();
int v = p.second;
// cout << v << " : " << dist[v] << endl;
if (vis[v])
continue;
ans++;
vis[v] = 1;
dfs_vis(v, v);
}
cout << ans << endl;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |