Submission #1106120

#	Time	Username	Problem	Language	Result	Execution time	Memory
1106120		ortsac	Cat in a tree (BOI17_catinatree)	C++17	0 / 100	3 ms	5140 KiB

catinatree

#include <bits/stdc++.h>

using namespace std;

#define pii pair<int, int>
#define fr first
#define se second

const int MAXN = 2e5 + 10;
vector<int> adj[MAXN];
int n, d;

pii calc(int node) {
    if (adj[node].empty()) {
        return {0, 1};
    }
    vector<pii> dec;
    int sum = 0;
    for (auto u : adj[node]) {
        dec.push_back(calc(u));
    }
    sort(dec.begin(), dec.end(), greater<pii>());
    int lst = 0x3f3f3f3f, ans = 0, mi = 0x3f3f3f3f;
    for (auto u : dec) {
        if ((lst + u.fr + 2) >= d) {
            ans += u.se;
            mi = min(lst, u.fr);
            lst = u.fr;
        } else {
            ans += (u.se - 1);
        }
    }
    mi++;
    if (mi == d) {
        lst = 0;
        ans++;
    }
    return {mi, ans};
}

int32_t main() {
    cin >> n >> d;
    for (int i = 1  ; i < n; i++) {
        int x;
        cin >> x;
        adj[x].push_back(i);
    }
    cout << calc(0).se << "\n";
}

Compilation message (stderr)

catinatree.cpp: In function 'std::pair<int, int> calc(int)':
catinatree.cpp:18:9: warning: unused variable 'sum' [-Wunused-variable]
   18 |     int sum = 0;
      |         ^~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 40
• Subtask 3: 0 / 49