Luogu 2279 [HNOI2003]消防局的设立 - 贪心

Description

给定一棵树形图， 建若干个消防站， 消防站能够覆盖到距离不超过2的点， 求最少需要建几个消防站才能覆盖所有点

Solution

从深度最深的点开始， 在它的爷爷节点上建， 每建一次都要把能覆盖的点都记录下来。

执行的次数就是答案。

Code

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define rd read()
 using namespace std;

 const int N = 1e3 + ;

 int n, f[N], id[N], vis[N], dep[N];
 int ans, head[N], tot;

 struct edge {
     int nxt, to;
 }e[N];

 int read() {
     int X = , p = ; char c = getchar();
     for(; c > '' || c < ''; c = getchar()) if(c == '-') p = -;
     for(; c >= '' && c <= ''; c = getchar()) X = X *  + c - '';
     return X * p;
 }

 void add(int u, int v) {
     e[++tot].to = v;
     e[tot].nxt = head[u];
     head[u] = tot;
 }

 int cmp(int a, int b) {
     return dep[a] > dep[b];
 }

 void dfs(int x) {
     if(!x) return;
     vis[x] = ;
     for(int i = head[x]; i; i = e[i].nxt) {
         int nt = e[i].to;
         vis[nt] = ;
     }
 }

 void col(int x) {
     vis[x] = ; vis[f[x]] = ;
     dfs(f[x]); dfs(f[f[x]]);
     x = f[f[x]];
     vis[x] = ; vis[f[x]] = ;
     dfs(f[x]);
     vis[f[f[x]]] = ;
     for(int i = head[x]; i; i = e[i].nxt) dfs(e[i].to);
 }

 int main()
 {
     n = rd;
     dep[] = ;
     for(int i = ; i <= n; ++i) {
         f[i] = rd;
         dep[i] = dep[f[i]] + ;
         add(f[i], i);
     }
     for(int i = ; i <= n; ++i) id[i] = i;
     sort(id + , id + n + , cmp);
     for(int i = ; i <= n; ++i) {
         if(vis[id[i]]) continue;
         ans++;
         col(id[i]);
     }
     printf("%d\n", ans);
 }