hdu6060 RXD and dividing

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6060

题意：
有一棵n个点的数，将2到n这n-1个点进行划分，分成k个点集，求每个点集并上1这个点后的最小斯坦纳树之和，要求这个和最大。

可以参考这两个博客：
http://blog.csdn.net/kkkkahlua/article/details/76562559
http://blog.csdn.net/your_eyes_see_sea/article/details/76578077

思路：
一条边ab对答案贡献了多少，取决于b的子树中的节点被分进了多少个点集，分进的点击越多，对答案的贡献越大。b子树上的点最多被分进min(k, son[b])个点集中。
枚举每条边aibi，求和aibi * min(k, son[bi])，就可以得到答案。

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

typedef long long ll;

struct Node
{
    int to, cost;
    Node(){}
    Node(int to, int cost):to(to), cost(cost){}
};


const int maxn = 1e6 + 10;
ll ans, deep;
int n, k, a, b, c;
bool vis[maxn];
int son[maxn];
vector<Node> edges[maxn];

void init(int n){
    ans = 0;
    deep = 0;
    memset(vis, 0, sizeof(vis));
    memset(son, 0, sizeof(son));
    for(int i=0; i<n+5; ++i) edges[i].clear();

}



void dfs(int fa, int cost){
    if(vis[fa]) return;
    vis[fa] = true;
    int len = edges[fa].size();
    // son[fa] = len;
    son[fa] = 1;
    for(int i=0; i<len; ++i){
        int to = edges[fa][i].to;
        if(!vis[to]){
            // ++deep;
            dfs(to, edges[fa][i].cost);
            // --deep;
            son[fa] += son[to];
        }
    }
    ans += (ll)cost*min(k, son[fa]);

    // cout<<deep<<endl;
    // cout<<cost<<" "<<son[fa]<<" fa is: "<<fa<<endl;
}   

int main(){
    // freopen("1005.in", "r", stdin);
    while(scanf("%d%d", &n, &k)!=EOF){
        init(n);
        for(int i=0; i<n-1; ++i){
            scanf("%d%d%d", &a, &b, &c);
            edges[a].push_back(Node(b, c));
            edges[b].push_back(Node(a, c));
        }
        // cout<<"test2"<<endl;
        dfs(1, 0);
        // cout<<"test1"<<endl;
        printf("%lld\n", ans);
    }
    return 0;
}

数据太大，在本地上跑不出官方提供的n是1e6的数据，dfs爆栈。标程是bfs。

bfs一遍，可以记录每个节点的父节点，再用循环跑一遍，将最下面一层的节点个数加到上一层，这样就可以求出每个节点的子孙个数。