树形DP+状态机模型

 AcWing 285. 没有上司的舞会

题意：

 思路：f[i][0]：表示从以i为根节点的子树中选且不选择i的方案。
f[i][1]：表示从以i为根节点的子树中选且包括i的方案。
f[i][0]=Σmax(f[j][0],f[j][1])，f[i][1]=Σf[j][0]；（j是i的字节点）。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=6010;
int happy[N],f[N][2];
int h[N],e[N],ne[N],idx;
bool root[N];//查找整棵树的根节点
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u)
{
	f[u][1]=happy[u];
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		dfs(j);
		f[u][1]+=f[j][0];
		f[u][0]+=max(f[j][0],f[j][1]);
	}
}
int main()
{
	int n;
	memset(h,-1,sizeof(h));
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&happy[i]);
	int x,y;
	for(int i=0;i<n-1;i++)
	{
		scanf("%d%d",&x,&y);
		root[x]=true;//x有父节点
		add(y,x);
	}
	int r=1;
	while(root[r]) r++;//找根节点
	dfs(r);
	printf("%d\n",max(f[r][1],f[r][0]));
	return 0;
}

323. 战略游戏（与上司的舞会类似）
题意：一棵树，每条边上至少要有一个士兵，求需要的最少的士兵数量。
思路：f[i][j]表示结点i的状态为j时的所有方案。
f[i][0]表示不选结点i，则f[i][0]+=f[son][1]
f[i][1]表示选结点i，则f[i][1]+=min(f[son][0],f[son][1])

#include<iostream>
#include<cstring>
using namespace std;
const int N=1510;
int e[N],ne[N],h[N],idx;
bool st[N];
int f[N][2];
int n;
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u)
{
	f[u][0]=0,f[u][1]=1;
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		dfs(j);
		f[u][0]+=f[j][1];
		f[u][1]+=min(f[j][0],f[j][1]);
	}
}
int main()
{
	while(~scanf("%d",&n))
	{
		memset(h,-1,sizeof(h));
        memset(st,0,sizeof(st));
        idx=0;
		for(int i=0;i<n;i++)
		{
			int a,b,s;
			scanf("%d:(%d)",&a,&s);
			while(s--)
			{
				scanf("%d",&b);
				st[b]=true;
				add(a,b);
			}
		}
		int root=0;
		while(st[root])//找根节点
			root++;
		dfs(root);
		printf("%d\n",min(f[root][0],f[root][1]));
	}
	return 0;
}

1077. 皇宫看守

题意：节点i能看到周围所有与它相连的结点，问至少在节点上放置多少个士兵花费最小。               这道题问的是在一堆点中至少选一个点，而“战略游戏”是在一条边里至少要选一个点。
思路：                                                                                                                                             状态表示：f[i][0]表示结点i被它的父节点看到的所有方案的最小值。
f[i][1]表示结点i被它的子节点看到的所有方案的最小值。
f[i][2]表示在结点i上放置士兵。
状态计算：f[i][0]=∑min(f[j][1],f[j][2])，i的子节点j就只能被j的子节点看到或在j这里放置士兵。
f[i][1]=min(f[k][2]+∑min(f[j][1],f[j][2]))，枚举被哪个子节点看到f[k][2]，剩下的子节点要么自己放置士兵，要么被它们的子节点看到。
f[i][2]=min(f[j][0],f[j][1],f[j][2])

#include<iostream>
#include<cstring>
using namespace std;
const int N=1510;
int h[N],e[N],ne[N],idx;
int f[N][3],w[N];
bool st[N];
int n;
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u)
{
	f[u][2]=w[u];//在自己这个点上放置士兵的花费
	int sum=0;
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		dfs(j);
		f[u][0]+=min(f[j][1],f[j][2]);
		f[u][2]+=min(f[j][0],min(f[j][1],f[j][2]));
		sum+=min(f[j][1],f[j][2]);
	}
	f[u][1]=1e9;
	for(int i=h[u];i!=-1;i=ne[i])//枚举被哪个子节点看到的花费最小
	{
		int j=e[i];
		f[u][1]=min(f[u][1],f[j][2]+sum-min(f[j][1],f[j][2]));
	}
}
int main()
{
	scanf("%d",&n);
	int a,b,c,cnt;
	memset(h,-1,sizeof(h));
	for(int i=0;i<n;i++)
	{
		scanf("%d%d%d",&a,&c,&cnt);
		w[a]=c;
		while(cnt--)
		{
			scanf("%d",&b);
			add(a,b);
			st[b]=true;
		}
	}
	int root=1;
	while(st[root]) root++;
	dfs(root);
	printf("%d\n",min(f[root][1],f[root][2]));
	return 0;
}