Luogu 2279 [HNOI2003]消防局的设立

Description

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

Solution

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

执行的次数就是答案。

Code

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

const int N = 1e3 + 5;

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 = 0, p = 1; char c = getchar();
    for(; c > '9' || c < '0'; c = getchar()) if(c == '-') p = -1;
    for(; c >= '0' && c <= '9'; c = getchar()) X = X * 10 + c - '0';
    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] = 1;
    for(int i = head[x]; i; i = e[i].nxt) {
        int nt = e[i].to;
        vis[nt] = 1;
    }
}

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

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