luogu2279_[HNOI2003]消防局的设立 贪心

传送门

不需要树形dp

关于深度排序

当前节点到最近的消防局（f[u]）>2时要建新的与u的上面（v）的上面(w)

同时w的上面和上面的上面也要更新f值

#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = a; i <= b; ++i)

const int N = 1007;

int n, a[N], id[N], dep[N], f[N];

inline bool cmp(const int &x, const int &y) {
    return dep[x] > dep[y];
}

int main() {
    scanf("%d", &n);
    
    rep(i, 1, n) id[i] = i;

    dep[1] = 1;
    rep(i, 2, n) {
        scanf("%d", &a[i]);
        dep[i] = dep[a[i]] + 1;
    }

    sort(id + 1, id + n + 1, cmp);

    memset(f, 0x3f, sizeof(f));

    int ans = 0;
    
    rep(i, 1, n) {
        int u = id[i], v = a[u], w = a[v];
        f[u] = min(f[u], min(f[v] + 1, f[w] + 2));
        if (f[u] > 2) {
            f[w] = 0, ans++;
            f[a[w]] = min(f[a[w]], 1), f[a[a[w]]] = min(f[a[a[w]]], 2);
        }
    }

    printf("%d\n", ans);

    return 0;
}