【HDU-2376】Average distance

題意：

給你一個樹，求樹上任意兩個點之間的距離的平均值

題解：

就是求出任意兩點之間的距離和然后除以邊數
”任意兩點之間的距離“ 和怎么算？
我們去考慮邊的貢獻情況：
對于邊(u,v)，u是v的父親節點，siz[v]表示以v為根的子樹的節點數量(包含v)，(n-siz[v]) 表示不在以v為根的子樹的節點數量，這兩部分中各選一個點，構成的連線都會經過(u,v)，n是總節點,那么答案就是這個邊的貢獻就是：ans=(n-siz[v]) * siz[v] * w

代碼：

#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() {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)));} }

總結

以上是生活随笔為你收集整理的【HDU-2376】Average distance的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。