题目描述
2020年,人类在火星上建立了一个庞大的基地群,总共有n个基地。起初为了节约材料,人类只修建了n-1条道路来连接这些基地,并且每两个基地都能够通过道路到达,所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话,我们称基地A到基地B的距离为d。
由于火星上非常干燥,经常引发火灾,人类决定在火星上修建若干个消防局。消防局只能修建在基地里,每个消防局有能力扑灭与它距离不超过2的基地的火灾。
你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时,消防队有能力及时扑灭火灾。
输入文件的第一行为n (n<=1000),表示火星上基地的数目。接下来的n-1行每行有一个正整数,其中文件第i行的正整数为a[i],表示从编号为i的基地到编号为a[i]的基地之间有一条道路,为了更加简洁的描述树状结构的基地群,有a[i]<i。
输出格式:
输出文件仅有一个正整数,表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。
输入
6
1
2
3
4
5
输出:
2
思路之一:
贪心地覆盖。每次寻找最大深度的节点,若未被覆盖则为了覆盖它且尽量不浪费,将其爷爷设为站点并更新父辈的距离。
代码:
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e3 + 5;
int n, f[maxn], ans;
int d[maxn], b[maxn], dis[maxn];
int main() {
scanf("%d", &n);
for (int i = 2; i <= n; i++) {
scanf("%d", &f[i]);
d[i] = d[f[i]] + 1;//d数组表示深度
}
for (int i = 0; i <= n; i++) {
b[i] = i;//排序用的
dis[i] = maxn;//该点到最近的消防站的距离
}
sort(b + 1, b + 1 + n, [](int x, int y){ return d[x] > d[y]; });
for (int i = 1; i <= n; i++) {
int cur = b[i], fa = f[cur], gf = f[fa];
dis[cur] = min(dis[cur], min(dis[fa] + 1, dis[gf] + 2));
if (dis[cur] > 2) {
ans++;
dis[gf] = 0;
dis[f[gf]] = min(dis[f[gf]], 1);
dis[f[f[gf]]] = min(dis[f[f[gf]]], 2);
}
}
return !printf("%d\n", ans);
}
点赞的时候,请宠溺一点