[洛谷]P2016 战略游戏 (#树形dp)

题目描述

Bob喜欢玩电脑游戏，特别是战略游戏。但是他经常无法找到快速玩过游戏的办法。现在他有个问题。

他要建立一个古城堡，城堡中的路形成一棵树。他要在这棵树的结点上放置最少数目的士兵，使得这些士兵能了望到所有的路。

注意，某个士兵在一个结点上时，与该结点相连的所有边将都可以被了望到。

请你编一程序，给定一树，帮Bob计算出他需要放置最少的士兵.

输入格式

第一行 N，表示树中结点的数目。

第二行至第N+1行，每行描述每个结点信息，依次为：该结点标号i，k(后面有k条边与结点I相连)。

接下来k个数，分别是每条边的另一个结点标号r1，r2，...，rk。

对于一个n(0<n<=1500)个结点的树，结点标号在0到n-1之间，在输入数据中每条边只出现一次。

输出格式

输出文件仅包含一个数，为所求的最少的士兵数目。

例如，对于如下图所示的树：

       0
1
2      3

答案为1（只要一个士兵在结点1上）。

输入输出样例

输入 #1复制

4
0 1 1
1 2 2 3
2 0
3 0

输出 #1复制

1

---

思路

突然发现树形dp的本质----每个节点都可以成为根节点，然后在这个节点搜索它的子树获取最优解。

令dp[i][0/1]为在节点i上放或不放士兵，0为不放，1为放。如果节点i上不放士兵，就意味着节点i的子节点必须要放士兵。则：

1.如果节点i不放士兵

dp[i][0]=dp[i][0]+dp[son][1]

其中son是i的子节点。

2.如果节点i放士兵

dp[i][1]=dp[i][1]+min(dp[son][0],dp[son][1])

答案为min(dp[root][0],dp[root][1])。

#include <stdio.h>
#include <iostream>
#include <memory.h>
#define maxn 1501
using namespace std;
int n,m,s,cnt,head[maxn],dp[maxn][2];
struct node
{
	int nxt,to;
}e[maxn<<1];
inline void add(int u,int v)
{
	e[++cnt].to=v;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs(int i,int fa)
{
	dp[i][1]=1,dp[i][0]=0;
	register int j;
	for(j=head[i];j;j=e[j].nxt)
	{
		int v(e[j].to);//子节点 
		if(v==fa) continue;//dfs多设一个父节点，判断节点i的子节点是否和父节点相等，相等的情况要排除 
		dfs(v,i);
		dp[i][0]+=dp[v][1];
		dp[i][1]+=min(dp[v][1],dp[v][0]);
	}
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		int u,v;
		cin>>u>>m;
		for(j=1;j<=m;j++)
		{
			cin>>v;
			add(u,v);
			add(v,u);
		}
	}
	dfs(0,-1);
	cout<<min(dp[0][1],dp[0][0])<<endl;
	return 0;
}