HDU6060-RXD and dividing 简单树形DP+贪心

传送门

题意：

给了一棵以1为原点，n个点的树，然后将其余的n-1个点分成k个部分，每个部分又都是以1为根的最小斯坦纳树，求k个最小斯坦纳树的边权最大总值。

思路：

1、由于是在一棵树上选边，所以其实不存在建最小斯坦纳树的问题；

2、贪心的思想，对于每一条边，方向为从父节点到子节点，其所能达到的最大贡献为，len * Max（k,son_node），其中 len 为边的长度，sum_node为子节点及其子树的总点数。每一条边的最大值即为全部的最大值。

#include <bits/stdc++.h>using namespace std;#define N 1000005#define LL long longint tot;struct edge{    int v,nxt,len;}e[N<<1];int head[N];void add(int a,int b,int c){    e[tot].v = b;    e[tot].nxt = head[a];    e[tot].len = c;    head[a] = tot++;}LL ans;int k;int dfs(int p,int pre){    int num = 0;    for(int i=head[p];~i;i=e[i].nxt){        int v = e[i].v;        if(v == pre) continue;        int cot = dfs(v,p);        ans += (LL)e[i].len * min(k,cot);        num += cot;    }      return num + 1;}int main(){    int n;    while(~scanf("%d%d",&n,&k)){        memset(head,-1,sizeof(head));        tot = 0;        for(int i=1;i<n;i++){            int a,b,c;            scanf("%d%d%d",&a,&b,&c);            add(a,b,c);            add(b,a,c);        }        ans = 0;        dfs(1,-1);        printf("%lld\n",ans);    }}