记录一个菜逼的成长。。
2017 Multi-University Training Contest - Team 3
题目链接
题目大意:
给你一棵
n
个点的树,将
求每个集合
f(S)
表示添加一些边使得
S
集合中的点连通,并且使得边的权值和最小
考虑每条边的贡献。
对于一条边
要使得这条边的贡献最大,要使以
x
为父节点的节点(包括
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
#define rep(i,l,r) for( int i = l; i <= r; i++ )
#define rep0(i,l,r) for( int i = l; i < r; i++ )
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define clr clear()
#define pb push_back
#define mp make_pair
#define fi first
#define se second
typedef long long LL;
typedef pair<int,int> PII;
const int INF = 0x3f3f3f3f;
const int maxn = 1000000 + 10;
vector<PII>edge[maxn];
int son[maxn];
LL ans;
int n,k;
void dfs(int u,int fa)
{
son[u] = 1;
for( int i = 0; i < edge[u].size(); i++ ){
int v = edge[u][i].fi,w = edge[u][i].se;
if(v == fa)continue;
dfs(v,u);
int sum = 0;
sum += son[v];
ans += (LL)w * min(k,sum);
son[u] += sum;
}
}
int main()
{
while(~scanf("%d%d",&n,&k)){
cl(son,0);
for( int i = 1; i <= n; i++ )edge[i].resize(0);
for( int i = 1; i < n; i++ ){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
edge[u].pb(mp(v,w));
edge[v].pb(mp(u,w));
}
ans = 0;
dfs(1,0);
printf("%lld\n",ans);
}
return 0;
}