一道难题Tree

题目背景

冴月麟和魏潇承是好朋友。

题目描述

冴月麟为了守护幻想乡，而制造了幻想乡的倒影，将真实的幻想乡封印了。任何人都无法进入真实的幻想乡了，但是她给前来救她的魏潇承留了一个线索。

她设置了一棵树（有根）。树的每一条边上具有割掉该边的代价。

魏潇承需要计算出割开这棵树的最小代价，这就是冴月麟和魏潇承约定的小秘密。

帮帮魏潇承吧。

注：所谓割开一棵有根树，就是删除若干条边，使得任何任何叶子节点和根节点不连通。

输入输出格式

输入格式：

输入第一行两个整数n，S表示树的节点个数和根。

接下来n-1行每行三个整数a、b、c，表示a、b之间有一条代价为c的边。

输出格式：

输出包含一行，一个整数，表示所求最小代价。

输入输出样例

输入样例#1：

4 11 2 1 1 3 11 4 1

输出样例#1：

3

输入样例#2：

4 11 2 32 3 13 4 2

输出样例#2：

1

说明

对于20%的数据，n <= 10

对于50%的数据，n <= 1000

对于100%的数据，n <= 100000

#include<iostream>#include<cstring>#include<algorithm>#define f(i,l,r) for(i=(l);i<=(r);i++)using namespace std;const int MAXN=200005;struct Edge{int v,w,nxt;}e[MAXN<<1];int head[MAXN],tot,dp[MAXN];inline void add(int u,int v,int w){e[tot].v=v;e[tot].w=w;e[tot].nxt=head[u];head[u]=tot++;}inline void dfs(int u,int fa){int i;for(i=head[u];~i;i=e[i].nxt){int v=e[i].v;if(v==fa) continue;dfs(v,u);if(dp[v]) dp[u]+=min(e[i].w,dp[v]);else dp[u]+=e[i].w;}}int main(){ios::sync_with_stdio(false);memset(head,-1,sizeof(head));int i,j,u,v,w,n,root;cin>>n>>root;f(i,1,n-1){cin>>u>>v>>w;add(u,v,w);add(v,u,w);}dfs(root,-1);cout<<dp[root]<<endl;return 0;}