[Luogu] P2279 [HNOI2003]消防局的设立

题目描述

2020年，人类在火星上建立了一个庞大的基地群，总共有n个基地。起初为了节约材料，人类只修建了n-1条道路来连接这些基地，并且每两个基地都能够通过道路到达，所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话，我们称基地A到基地B的距离为d。

由于火星上非常干燥，经常引发火灾，人类决定在火星上修建若干个消防局。消防局只能修建在基地里，每个消防局有能力扑灭与它距离不超过2的基地的火灾。

你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时，消防队有能力及时扑灭火灾。

输入输出格式

输入格式：

输入文件名为input.txt。

输入文件的第一行为n （n<=1000），表示火星上基地的数目。接下来的n-1行每行有一个正整数，其中文件第i行的正整数为a[i]，表示从编号为i的基地到编号为a[i]的基地之间有一条道路，为了更加简洁的描述树状结构的基地群，有a[i]<i。

输出格式：

输出文件名为output.txt

输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

题目解析

一眼树形DP，然而事实上可以贪心。

枚举每一个叶节点，从里面找出深度最大的，向周围距离小于4的区域扩展，作为一个消防局的控制范围。再把没被访问过且没有儿子的节点也作为叶节点，直到结束。

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;

const int MAXN = 1005;

struct Edge {
    int nxt;
    int to;
} l[MAXN << 1];
struct Node {
    int deep,id;
    bool leaf;
    friend bool operator < (Node x,Node y) {
        return x.deep < y.deep;
    }
} a[MAXN];

int n,ans;
int head[MAXN],cnt;
bool vis[MAXN];
priority_queue<Node> q;

void dfs1(int x,int from) {
    a[x].deep = a[from].deep + 1;
    for(int i = head[x];i;i = l[i].nxt) {
        if(l[i].to == from) continue;
        a[x].leaf = true;
        dfs1(l[i].to,x);
    }
    if(!a[x].leaf) q.push(a[x]);
    return;
}

void dfs2(int x,int point,int from) {
    if(!point) return;
    if(point == 1) {
        for(int i = head[x];i;i = l[i].nxt) {
            if(vis[l[i].to]) continue;
            a[l[i].to].leaf = false;
            q.push(a[l[i].to]);
        }
        return;
    }
    for(int i = head[x];i;i = l[i].nxt) {
        if(l[i].to == from) continue;
        vis[l[i].to] = true;
        dfs2(l[i].to,point-1,x);
    }
}

inline void add(int x,int y) {
    cnt++;
    l[cnt].nxt = head[x];
    l[cnt].to = y;
    head[x] = cnt;
    return;
}

int main() {
    scanf("%d",&n);
    int x;
    for(int i = 2;i <= n;i++) {
        scanf("%d",&x);
        add(i,x),add(x,i);
    }
    for(int i = 1;i <= n;i++) a[i].id = i;
    dfs1(1,0);
    int cur;
    while(!q.empty()) {
        cur = q.top().id;
        q.pop();
        if(vis[cur]) continue;
        vis[cur] = true;
        dfs2(cur,5,0);
        ans++;
    }
    printf("%d\n",ans);
    return 0;
}