Bonsai HDU - 3452（树形DP）

Bonsai

 题目链接：HDU - 3452

题意：给出一棵树，已知树根，边权，要求删掉一些边，使得所有叶子都不能再和树根有联通路径，删掉的边的边权之和最小是多少；

思路：不考虑限制条件，我们可以直接将与树根直接相连的边删掉，也可以将叶子上的连边删掉，这是两种最简单的方法，但是未必使得删掉的边权之和最小；

如上图我们可以看到，删除2-4连边的效果和删除4-8， 4-9连边的效果相同；

删除1-3连边的效果等同于删除3-6，3-7连边；

所以说删边就只有两种情况了，要么删i节点与son的两边，要么删i节点与father的了连边，取最小值；

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
struct node{
	int v, w, nxt;
	node(){}
	node(int v0, int w0, int n){
		v=v0, w=w0, nxt=n;
	}
}edge[2100];
int cnt, head[1100];
void add(int u, int v, int w){
	edge[cnt]=node(v, w, head[u]);
	head[u]=cnt++;
}
int dp[1100];
int dfs(int u, int fa){
	dp[u]=0;
	int flag=0;
	for(int i=head[u]; i!=-1; i=edge[i].nxt){
		int v=edge[i].v, w=edge[i].w;
		if(v==fa) continue;
		flag=1;
		dp[u]+=min(w, dfs(v, u));
	}
	return flag?dp[u]:INF;
}
int main(){
	int n, r;
	while(scanf("%d%d", &n, &r), n||r){
		memset(head, -1, sizeof(head));
		cnt=0;
		for(int i=1; i<n; i++){
			int u, v, w;
			scanf("%d%d%d", &u, &v, &w);
			add(u, v, w);
			add(v, u, w);
		}
                if(n<=1) printf("0\n");
		else printf("%d\n", dfs(r, r));
	}
	return 0;
}