HDU #2376. Average distance

题意

问树上路径长度的期望值

题解

dfs求出每条边左右连接的点数,统计答案,最后*2/n/(n+1)

调试记录

#include <cstdio>
#include <cstring>
#define int long long
const int maxn = 1e4 + 5;
using namespace std;

struct node{
	int to, next, l;
}e[maxn << 1];
int head[maxn], tot;
void addedge(int u, int v, int l){
	e[++tot].to = v, e[tot].next = head[u], e[tot].l = l;
	head[u] = tot;
}

long long ans; int size[maxn], n;
void dfs(int cur, int fa){
	size[cur] = 1;
	for (int i = head[cur]; i; i = e[i].next){
		if (e[i].to != fa){
			dfs(e[i].to, cur);
			size[cur] += size[e[i].to];
			ans += e[i].l * size[e[i].to] * (n - size[e[i].to]);
		}
	}
}

int T;
signed main(){
	scanf("%lld", &T);
	while (T--){
		scanf("%lld", &n);
		memset(e, 0, sizeof e);
		memset(head, 0, sizeof head);
		memset(size, 0, sizeof size);
		tot = 0; ans = 0;
		for (int u, v, l, i = 1; i < n; i++){
			scanf("%lld%lld%lld", &u, &v, &l); u++, v++;
			addedge(u, v, l), addedge(v, u, l);
		}
		dfs(1, 0);
		printf("%.6lf\n", 2.0 * ans / n / (n - 1));
	}
	
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值