（树形DP）【HDU-2376】Average distance

最新推荐文章于 2021-05-18 18:40:24 发布

Samson_Ho

最新推荐文章于 2021-05-18 18:40:24 发布

阅读量173

点赞数

分类专栏： DP动态规划

本文链接：https://blog.csdn.net/suiguia/article/details/82629147

版权

DP动态规划专栏收录该内容

9 篇文章 0 订阅

订阅专栏

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

题目大意就是给一棵树，求出树上任意两点间距离的平均值。等价于求出任意两点的距离的和，再除以边数就得到答案了。

求距离和的话，一条边（A,B）的贡献值 = 经过A的次数*经过B的次数*AB这条边的边权

/*
* @Author: SamsonHo
* @Date:   2018-08-28-16.04
* @URL:http://acm.hdu.edu.cn/showproblem.php?pid=2376
* @Note:题目大意就是给一棵树，求出树上任意两点间距离的平均值。等价于求出任意两点的距离的和，再除以边数就得到答案了。

求距离和的话，一条边（A,B）的贡献值 = 经过A的次数*经过B的次数*AB这条边的边权
*/
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long LL;
const int MAXN = 1e4+10;
int n;
vector<pair<int,int> > e[MAXN];
LL dp[MAXN],sum[MAXN];

void dfs(int cur,int father)
{
    sum[cur] = 1;
    int len = e[cur].size();
    for(int i = 0; i < len; ++i)
    {
        int v = e[cur][i].first;
        if(v == father)    continue;
        dfs(v,cur);
        sum[cur] += sum[v];
        dp[cur] += dp[v] + sum[v]*(n-sum[v])*(e[cur][i].second);
    }
    return;
}

int main(void)
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i = 0; i <= n; ++i)
        {
            e[i].clear();
            dp[i] = sum[i] = 0;
        }
        int u,v,w;
        for(int i = 0; i < n-1; ++i)
        {
            scanf("%d%d%d",&u,&v,&w);
            e[u].push_back(make_pair(v,w));
            e[v].push_back(make_pair(u,w));
        }
        dfs(1,-1);
        printf("%.11f\n",(double)(dp[1]/(n*(n-1)/2.0)));
    }
}