HDU 6060 RXD and dividing

题目链接：

题解：

代码：

#include <map>
#include <cmath>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
typedef long long ll;
#define inf 0x3f3f3f3f
const int maxn = 1e6+10;
vector<ll>p[maxn];
map<ll,ll> mp[maxn];
ll visited[maxn];
ll n,k;
ll ans;

void dfs(ll x,ll f,ll w)
{
visited[x]=1;
for(int i=0;i<p[x].size();i++)
{
ll y=p[x][i];
if(y==f)
continue;
dfs(y,x,mp[x][y]);
visited[x]+=visited[y];
}
ans+=w*min(k,visited[x]);
}

int main()
{
while(scanf("%lld%lld",&n,&k)!=EOF)
{
for(int i=1;i<=n;i++)
{
p[i].clear();
mp[i].clear();
}
for(int i=1;i<n;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
p[a].push_back(b);
p[b].push_back(a);
mp[a][b]=c;
mp[b][a]=c;
}
met(visited,0);
ans=0;
dfs(1,0,0);
printf("%lld\n",ans);

}
}