[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
输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。
输入输出样例
输入样例#1：
6
1
2
3
4
5
输出样例#1：
2

题解

• 总体思路：贪心 + 图的遍历 +搜索。
• 我们可以先预处理出每个节点的深度，然后按照深度大小进行排序，求深度的过程我们可以用$bfs$，而在$bfs$的过程中我们可以开一个栈，我们的遍历顺序是从上向下的，即我们先遍历的是深度小的。
• 那么栈又是一个先进后出的数据结构，那么我们在把遍历到的节点加入栈时栈顶的元素是深度最大的节点的编号，这样就省去了我们排序的过程。
• 接着我们枚举栈中的每个节点，从节点的爷爷节点开始往下搜索即可。（因为爷爷节点到此节点的深度为2，符合题目要求。）

code

#include <bits/stdc++.h>
const int maxn = 1e3 + 100;
const int inf = 0x3f3f3f3f;
typedef long long LL;
using namespace std;

template <typename T>
inline void read(T &s) {
    s = 0;
    T w = 1, ch = getchar();
    while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
    while (isdigit(ch))  { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
    s *= w;
}

template<typename T>
inline void write(T s) {
	if (s < 0) putchar('-'), s = -s;
	if (s > 9) write(s / 10);
	putchar(s % 10 + '0');
}

int n;
int tot;
int ans;
int fa[maxn];
int lin[maxn];
int dep[maxn];
int nex[maxn << 1];
int ver[maxn << 1];
bool vis[maxn];
queue <int> q;
stack <int> s;

inline void add(int from, int to) {
	ver[++tot] = to;
	nex[tot] = lin[from];
	lin[from] = tot;
}

void bfs(int k) {
	q.push(k);
	s.push(k);
	dep[k] = 1;
	while (!q.empty()) {
		int u = q.front(); q.pop();
		for (int i = lin[u]; i; i = nex[i]) {
			int v = ver[i];
			if (v == fa[u]) continue;
			dep[v] = dep[u] + 1;
			q.push(v);
			s.push(v);
		}
	}
}

void dfs(int now, int depth) {
	if (depth > 2) return ;
	vis[now] = true;
	for (int i = lin[now]; i; i = nex[i]) {
		int v = ver[i];
		dfs(v, depth + 1); 
	}
}

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

int main() {
	read(n);
	for (int i = 2; i <= n; ++i) {
		read(fa[i]);
		add(i, fa[i]);
		add(fa[i], i);
	}
	bfs(1);
	work();
	write(ans);
	return 0;
}