HDU 3452 Bonsai

#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<map>
#include<vector>
#include<string>
#include<algorithm>
#include<iostream>
using namespace std;

int n,root;
const int maxn=1000+10;
struct Edge
{
int to;
int val;
Edge (int u,int v){to=u;val=v;}
};
vector<Edge>tree[maxn];
bool vis[maxn];
int dp[maxn];

void init()
{
for(int i=0;i<=n;i++) tree[i].clear();
memset(vis,0,sizeof vis);
memset(dp,-1,sizeof dp);
}

void read()
{
for(int i=1;i<=n-1;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
Edge e1(v,w); Edge e2(u,w);
tree[u].push_back(e1);
tree[v].push_back(e2);
}
}

void dfs(int now,int lim)
{
bool fail=1;
for(int i=0;i<tree[now].size();i++)
if(!vis[tree[now][i].to]) fail=0;

if(fail) return;

bool x=0;
int sum=0;

for(int i=0;i<tree[now].size();i++)
{
if(!vis[tree[now][i].to])
{
vis[tree[now][i].to]=1;
dfs(tree[now][i].to,lim);

if(dp[tree[now][i].to]!=-1)
sum=sum+min(tree[now][i].val,dp[tree[now][i].to]);
else
{
if(tree[now][i].val<=lim) sum=sum+tree[now][i].val;
else x=1;
}
}
}
if(!x) dp[now]=sum;
}

bool work(int lim)
{
memset(dp,-1,sizeof dp);
memset(vis,0,sizeof vis);
vis[root]=1;
dfs(root,lim);
if(n==1) dp[root]=0;
printf("%d\n",dp[root]);
return 1;
}

int main()
{
while(~scanf("%d%d",&n,&root))
{
if(!n&&!root) break;
init();
read();
work(2000);
}
return 0;
}