关闭

(2017多校训练第三场)HDU - 6060 RXD and dividing dfs

标签: HDU 6060RXD and dividingdfs
71人阅读 评论(0) 收藏 举报
分类:

官方题解:

2∼n2\sim n把1看成整棵树的根. 问题相当于把2n每个点一个[1,k][1, k][1,k]的标号. 然后根据最小斯坦纳树的定义, (x,fax)(x, fa_x)(x,fax) 这条边的贡献是 x 子树内不同标号的个数目difidif_idifi. 那么显然有difi≤min(k,szi)dif_i\leq min(k, sz_i)difimin(k,szi)szisz_iszi表示子树大小. 可以通过构造让所有difidif_idifi都取到最大值. 所以答案就是∑x=2nw[x][fax]∗min(szx,k)\sum_{x = 2}^{n}{w[x][fa_x] * min(sz_x, k)}x=2nw[x][fax]min(szx,k)时间复杂度为O(n)。O(n)O(n)位O(n)

代码如下:

#include <bits/stdc++.h>

using namespace std;

typedef long long int LL;
const int MOD = 1e9 + 7;
const int MAX_N = 1e6 + 5;
const int INF = 0x3f3f3f3f;
int n, k;
struct edge
{
    int to;
    int cost;
};
vector<edge> G[MAX_N];
int siz[MAX_N];
int pre[MAX_N];

void dfs(int u, int fa)
{
    siz[u] = 1;
    for (unsigned i = 0; i < G[u].size(); i++)
    {
        int v = G[u][i].to;
        int w = G[u][i].cost;
        if (v == fa)
            continue;
        dfs(v, u);
        pre[v] = w;
        siz[u] += siz[v];
    }
}

int main()
{
    //freopen("test.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    cin.sync_with_stdio(false);
    while (cin >> n >> k)
    {
        for (int i = 0; i < MAX_N; i++)
            G[i].clear();
        for (int i = 1; i < n; i++)
        {
            int u, v, cost;
            cin >> u >> v >> cost;
            G[u].push_back({v, cost});
            G[v].push_back({u, cost});
        }
        dfs(1, -1);
        LL ans = 0;
        for (int i = 2; i <= n; i++)
            ans += (LL)pre[i] * min(siz[i], k);
        cout << ans << endl;
    }
    return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:21522次
    • 积分:1682
    • 等级:
    • 排名:千里之外
    • 原创:151篇
    • 转载:2篇
    • 译文:0篇
    • 评论:26条