【BZOJ1217】[HNOI2003] 消防局的设立(树形dp,贪心)

本文探讨了在火星基地群中,如何通过算法确定最少消防局数量,以确保所有基地在发生火灾时都能得到及时扑灭。采用树形DP与贪心策略相结合的方法,通过对基地间的距离和树状结构进行分析,得出最优布局方案。

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

题目

Description

2020年,人类在火星上建立了一个庞大的基地群,总共有n个基地。起初为了节约材料,人类只修建了n-1条道路来
连接这些基地,并且每两个基地都能够通过道路到达,所以所有的基地形成了一个巨大的树状结构。如果基地A到
基地B至少要经过d条道路的话,我们称基地A到基地B的距离为d。由于火星上非常干燥,经常引发火灾,人类决定
在火星上修建若干个消防局。消防局只能修建在基地里,每个消防局有能力扑灭与它距离不超过2的基地的火灾。
你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时,消防队有能力及时扑灭火灾

Input

第一行为n,表示火星上基地的数目。N<=1000
接下来的n-1行每行有一个正整数,其中文件第i行的正整数为a[i],表示从编号为i的基地到编号为a[i]的基地之间有一条道路,
为了更加简洁的描述树状结构的基地群,有a[i] < i
Output
仅有一个正整数,表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

Sample Input

6
1
2
3
4
5

Sample Output

2

原题传送门

思路

树形dp可以做,可我用的贪心。
每隔4个点放是最优的。
特殊处理一下叶子节点就好了。

代码

#include <bits/stdc++.h>

using namespace std;
const int M = 500000;
struct node { int x, y, nxt; } a[M];
int len, la[M], ans, f[M], n;

inline int read() {
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {
        if(ch == '-') f = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        x = (x << 1)+(x << 3)+(ch ^ 48);
        ch = getchar();
    }
    return x*f;
}

void adde(int x, int y) {
    a[++len].x = x;
	a[len].y = y;
    a[len].nxt = la[x];
	la[x] = len;
}

void treedp(int x,int fa) {
    int maxn = -1292371547, minn = -maxn;
    for (int k = la[x]; k; k = a[k].nxt) {
        int y = a[k].y;
        if (y != fa) {
            treedp(y, x);
            maxn = max(maxn, f[y]);
			minn = min(minn, f[y]);
        }
    }
    if(maxn+minn <= 3) f[x] = minn+1;
    else f[x] = maxn+1;
    if (maxn == -1292371547) f[x] = 3;
    if (f[x] == 5) f[x] = 0, ans++;
    if (f[x] >= 3 && fa == -1) ans++; 
}

int main() {
	ios::sync_with_stdio(false);
	n = read();
    for (int i = 1; i < n; i++) {
        int x = read();
        adde(i+1,x), adde(x,i+1);
    }
    treedp(1, -1);
    cout << ans << endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值