洛谷 P2279 [HNOI2003]消防局的设立（树形DP 或贪心）

最新推荐文章于 2024-04-09 09:22:59 发布

原创最新推荐文章于 2024-04-09 09:22:59 发布 · 380 阅读

1 ·

CC 4.0 BY-SA版权

C++学习专栏收录该内容

87 篇文章

订阅专栏

博客围绕火星基地消防局建设问题展开。2020年火星有n个基地呈树状结构，每个消防局能扑灭距离不超2的基地火灾。给出输入输出格式，输入为基地数目及道路连接信息，输出为至少需建的消防局数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

任重而道远

题目描述

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

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

输入输出样例

输入样例#1： 复制

输出样例#1： 复制

AC代码：

//贪心
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
using namespace std;

const int N = 1e3 + 10;
struct Edge {
    int tov, nxt;
}e[N << 1];
queue <int> q;
stack <int> st;
int head[N << 1], vis[N], fa[N];
int n, num, root = 1, ans;

void add_edge (int u, int v) {
    num++;
    e[num].tov = v;
    e[num].nxt = head[u];
    head[u] = num;
}

void bfs () {
    q.push (root);
    st.push (root);
    while (!q.empty ()) {
        int n = q.front ();
        q.pop ();
        for (int i = head[n]; i; i = e[i].nxt) {
            int v = e[i].tov;
            if (v == fa[n]) continue;
            q.push (v);
            st.push (v);
        }
    }
}

void dfs (int u, int dis) {
    if (dis > 2) return;
    vis[u] = true;
    for (int i = head[u]; i; i = e[i].nxt) {
        int v = e[i].tov;
        dfs (v, dis + 1);
    }
}

void solve () {
    while (!st.empty ()) {
        int u = st.top ();
        st.pop ();
        if (!vis[u]) {
            dfs (fa[fa[u]], 0);
            ans++;
        }
    }
}

int main () {
    scanf ("%d", &n);
    for (int i = 2; i <= n; i++) {
        int u;
        scanf ("%d", &u);
        add_edge (u, i);
        add_edge (i, u);
        fa[i] = u;
    }
    bfs ();
    solve ();
    cout << ans;
}