题目描述
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上)。
输入样例
4
0 1 1
1 2 2 3
2 0
3 0
输出样例
1
思路
- 树形dp,开一个dp数组,用u表示父节点,v表示子节点,dp[u][0]表示u节点不放,dp[u][1]表示u节点放置士兵
- 然后找状态转移方程,若父节点不放士兵,那么子节点必定要放置士兵,dp[u][0]=dp[u][0]+dp[v][1]
- 若父节点放士兵,那么子节点可以放置士兵也可不放,dp[u][1]=dp[u][1]+min(dp[v][0],dp[v][1])
#include<bits/stdc++.h>
#define read() freopen("input.txt","r",stdin);
#define write() freopen("output.txt","w",stdout);
using namespace std;
const int maxn = 1e5+10;
vector<int>g[maxn];
int dp[maxn][2];
void dfs(int u){
dp[u][0]=0;
dp[u][1]=1;
for( int i=0; i<g[u].size(); i++ ){
int v=g[u][i];
dfs(v);
dp[u][0]+=dp[v][1];
dp[u][1]+=min(dp[v][0],dp[v][1]);
}
}
int main()
{
int n;scanf("%d",&n);
for( int i=0; i<n; i++ ){
int u,k,v;
scanf("%d %d",&u,&k);
for( int j=0; j<k; j++ ) scanf("%d",&v),g[u].push_back(v);
dfs(0);
}
printf("%d",min(dp[0][0],dp[0][1]));
return 0;
}