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 = ;

 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, , n) id[i] = i;

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

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

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

     int ans = ;

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

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

     return ;
 }