标签：distance HDU int siz Average cnt dfs edge head

题目链接：HDU 2376 Average distance

题目大意：
计算一棵树中任意两点之间的距离的平均值。

题解：
如果暴力枚举两点再求距离是显然会超时的。

转换一下思路，我们可以对每条边，求所有可能的路径经过此边的次数：设这条边两端的点数分别为\(a\)和\(b\)，那么这条边被经过的次数就是\(a\times b\)，它对总的距离和的贡献就是\(a\times b \times edge[i].len\)，我们把所有边的贡献求总和，再除以总路径数\(\frac{n\times (n-1)}{2}\)，即为最后所求。

每条边两端的点数的计算，实际上是可以用一次\(dfs\)解决的。任取一点为根，在\(dfs\)的过程中，对每个点\(k\)记录其子树包含的点数（包括其自身），设点数为\(siz[k]\)，则\(k\)的父亲一侧的点数即为\(n-siz[k]\)。

#include <cstring>
#include <iomanip>
#include <iostream>
using namespace std;

int siz[10010], n, t, cnt, head[10010];
long long ans;
struct Edge {
    int v, len, next;
} edge[20010];

void addEdge(int u, int v, int w) {
    edge[++cnt].v = v;
    edge[cnt].len = w;
    edge[cnt].next = head[u];
    head[u] = cnt;
}

void dfs(int u, int pre) {
    siz[u] = 1;
    for (int i = head[u]; ~i; i = edge[i].next) {
        int v = edge[i].v;
        if (v == pre) {
            continue;
        }
        dfs(v, u);
        siz[u] += siz[v];
        ans += 1ll * siz[v] * (n - siz[v]) * edge[i].len;
    }
}

int main() {
    cin >> t;
    while (t--) {
        cnt = 0;
        memset(head, -1, sizeof(head));
        cin >> n;
        for (int i = 1, a, b, d; i < n; ++i) {
            cin >> a >> b >> d;
            addEdge(a, b, d);
            addEdge(b, a, d);
        }
        ans = 0;
        dfs(0, -1);
        cout << fixed << setprecision(6) << (double)ans / (n * (n - 1) / 2) << endl;
    }
    return 0;
}

标签：distance,HDU,int,siz,Average,cnt,dfs,edge,head
来源： https://www.cnblogs.com/IzumiSagiri/p/15059662.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。