消防局的设立（树形DP）

消防局的设立

2020 年，人类在火星上建立了一个庞大的基地群，总共有 n 个基地。起初为了节约材料，人类只修建了 n-1 条长为 1 的道路来连接这些基地，并且每两个基地都能够通过道路到达，所以所有的基地形成了一个巨大的树状结构。

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

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

输入格式

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

输出格式

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

解法

这是一道稍复杂的树形DP，对于每个点，我定义了5个状态，分别为dp[ i ][ 1 ~ 5 ]，表示为 i 点提供消防服务的是 i 的①爷爷/兄弟、②父亲、③自己、④一个儿子、⑤一个孙子。

具体这样计算：

dfs(int x) {
	dp[x][3] = 1;
	int dpp1 = 0x7f7f7f7f;
	int dpp2 = 0x7f7f7f7f;
	枚举儿子 {
		dfs(儿子);
		int y = 儿子;
		dp[x][1] += min(dp[y][3],min(dp[y][4],dp[y][5]));
		dp[x][2] += min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5])));
		dp[x][3] += min(dp[y][2],min(dp[y][3],min(dp[y][4],dp[y][5])));
		dp[x][4] += min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5])));
		dpp1 = min(dpp1,dp[y][3] - min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5]))));
		dp[x][5] += min(dp[y][3],min(dp[y][4],dp[y][5]));
		dpp2 = min(dpp2,dp[y][4] - min(dp[y][3],min(dp[y][4],dp[y][5])));
	}
	dp[x][4] += dpp1;
	dp[x][5] += dpp2;
}

可见此题的状态转移方程很复杂。

CODE

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#define LL long long
using namespace std;

LL read() {
	LL f = 1,x = 0;char s = getchar();
	while(s < '0' || s > '9') {if(s == '-')f = -1;s = getchar();}
	while(s >= '0' && s <= '9') {x = x * 10 + s - '0';s = getchar();}
	return x * f;
}//读入优化
LL n,m,i,j,s,o,k;
vector<int> g[1005];//树
int dp[1005][10]; 
LL min(LL a,LL b) {
	return a < b ? a : b;
}
void dfs(int x,int fa) {
	dp[x][1] = dp[x][2] = dp[x][3] = dp[x][4] = dp[x][5] = 0;
	dp[x][3] = 1;
	int dpp1 = 0x7f7f7f7f;
	int dpp2 = 0x7f7f7f7f;
	for(int i = 0;i < g[x].size();i ++) {
		if(g[x][i] != fa) {
			dfs(g[x][i],x);
			int y = g[x][i];
            //以下是核心部分
			dp[x][1] += min(dp[y][3],min(dp[y][4],dp[y][5]));
			dp[x][2] += min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5])));
			dp[x][3] += min(dp[y][2],min(dp[y][3],min(dp[y][4],dp[y][5])));
			dp[x][4] += min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5])));
			dpp1 = min(dpp1,dp[y][3] - min(dp[y][1],min(dp[y][3],min(dp[y][4],dp[y][5]))));
			dp[x][5] += min(dp[y][3],min(dp[y][4],dp[y][5]));
			dpp2 = min(dpp2,dp[y][4] - min(dp[y][3],min(dp[y][4],dp[y][5])));
		}
	}
	dp[x][4] += dpp1;
	dp[x][5] += dpp2;
	return ;
}
int main() {
	n = read();
	for(int i = 2;i <= n;i ++) {
		s = read();
		g[i].push_back(s);
		g[s].push_back(i);
	}
	dfs(1,0);
	printf("%d\n",min(dp[1][3],min(dp[1][4],dp[1][5])));
	return 0; 
}