1、我的第一道树形DP~感觉对树的定义还是不熟,题目都说得那么直白了。。。恰好n-1条边,每两点有且仅有一条通路。。。不看题解根本不知道是树。
2、正好趁这道题熟悉了一下vector的用法,注意clear是不会改变元素数量的。。。
3、WA了好几次,原来是INF设得太大了(题目中涉及累加操作,容易溢出),要注意。
4、dp【i】【0】表示覆盖子树且i作为根有一个电台,dp【i】【1】表示覆盖子树且i作为根没有电台,dp【i】【2】表示覆盖除i结点外的子树且i作为根没有电台。
5、计算dp【u】【1】时,表达“至少一个”这个意思的时候,非常巧妙。
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
const int INF=1000000;
vector<int> g[10010];
int dp[10010][3];
void dfs(int u,int fa){
int sz=g[u].size();
int sum=0;
dp[u][0]=1;
for(int i=0;i<sz;i++){
int v=g[u][i];
if(v==fa) continue;
dfs(v,u);
dp[u][0]+=min(dp[v][0],min(dp[v][1],dp[v][2]));
sum+=min(dp[v][0],dp[v][1]);
dp[u][2]+=dp[v][1];
}
dp[u][1]=INF;
if(sz==1&&u!=1) return;
for(int i=0;i<sz;i++){
int v=g[u][i];
if(v==fa) continue;
dp[u][1]=min(dp[u][1],dp[v][0]+sum-min(dp[v][0],dp[v][1]));
}
}
int main(){
int u,v;
while(scanf("%d",&n)==1){
for(int i=1;i<=n;i++){
while(!g[i].empty()) g[i].pop_back();
}
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
memset(dp,0,sizeof(dp));
dfs(1,0);
printf("%d\n",min(dp[1][0],dp[1][1]));
}
return 0;
}