Submission #1514217

---

Source Code Expand

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

constexpr int INF = 1e9;

using graph = vector<vector<int>>;

int N, K;

void dfs(int v, graph const& g, int depth, int& res) {
    res += depth > K;
    depth %= (K + 1);
    for(auto to : g[v]) {
        dfs(to, g, depth + 1, res);
    }
}

int main() {
    cin >> N >> K;
    graph g(N);
    int res = 0;
    for(int i = 0; i < N; ++i) {
        int a;
        cin >> a;
        a--;
        if(i == 0) {
            res += a != 0;
        } else {
            g[a].push_back(i);
        }
    }
    dfs(0, g, 0, res);
    cout << res << endl;
}

Submission Info

Judge Result