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

洛谷传送门

BZOJ传送门

题目描述

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

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

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

输入输出格式

输入格式：

输入文件名为input.txt。

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

输出格式：

输出文件名为output.txt

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

输入输出样例

输入样例#1：

6
1
2
3
4
5

输出样例#1：

2

解题分析

直接按深度从大到小贪心取即可， 维护每个点到最近的消防站的距离。

代码如下：

#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define MX 1050
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
int head[MX], fat[MX], dis[MX], dep[MX], ord[MX];
IN bool cmp (R int x, R int y) {return dep[x] > dep[y];}
struct Edge {int to, nex;} edge[MX << 1];
int dot, cnt;
int main(void)
{
	R int fa, grand, ans = 0;
	in(dot); dis[1] = dot; ord[1] = 1; dis[0] = dot;
	for (R int i = 2; i <= dot; ++i)
	in(fat[i]), dis[i] = dot, dep[i] = dep[fat[i]] + 1, ord[i] = i;
	std::sort(ord + 1, ord + 1 + dot, cmp);
	for (R int i = 1; i <= dot; ++i)
	{
		fa = fat[ord[i]], grand = fat[fa];
		dis[ord[i]] = std::min(dis[ord[i]], std::min(dis[fa] + 1, dis[grand] + 2));
		if(dis[ord[i]] > 2)
		{
			dis[ord[i]] = 2; ++ans; dis[grand] = 0;
			dis[fat[grand]] = std::min(dis[fat[grand]], 1), dis[fat[fat[grand]]] = std::min(dis[fat[fat[grand]]], 2);
		}
	}
	printf("%d", ans);
}