hdu6446(树上dp)

给出一颗树，n个结点，n个结点有n！种排列组合，假设其中一种是1 3 5 2 4，那么就要求出1-3,3-5,5-2,2-4的路径总和，问所有排列组合的路径总和。

先考虑路径重复的问题，有n!中排列组合，每种组合有(n-1)条路径，不同路径数是n*(n-1)/2，那么每条路径重复次数就是2*(n-1)!

接下来就是求不同路径的长度总和。。。

本着大胆猜想的原则。。我猜测路径长度总和与路径无关，于是把整个树拉成直线。。。。WA+1

接下来写树上dp。。。WA+1

参考这个博客才知道怎么求树上路径总和的。。树形dp（路径总和）

计算过程中没有乘1LL。。。WA+1

 

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;
typedef long long ll;
const int mx = 1e5+5;
const int inf = 0x3f3f3f3f;
const long long mod = 1e9+7;
int n, son[mx];
ll p[mx], val[mx], ans;
bool vis[mx];

vector <pair<int, ll> > mat[mx];

void dfs(int u, int fa) {
	son[u] = 1;
	int len = mat[u].size();
	for (int i = 0; i < len; i++) {
		int v = mat[u][i].first;
		int w = mat[u][i].second;
		if (v == fa) continue;
		dfs(v, u);
		son[u] += son[v];
		//printf("son[%d] = %d\n", u, son[u]);
		val[u] += val[v] + 1LL * w * son[v] % mod * (n-son[v]) % mod;
		val[u] %= mod;
	}
}

void init() {
	for (int i = 0; i < mx; i++)
		mat[i].clear();
	memset(val, 0, sizeof(val));
	memset(vis, false, sizeof(vis));
	memset(son, 0, sizeof(son));
	ans = 0;
}

int main() {
	p[0] = 1;
	p[1] = 1;
	for (int i = 2; i < mx; i++) {
		p[i] = p[i-1]*i % mod;
	}
	while (scanf("%d", &n) != EOF) {
		init();
		int u, v, w;
		for (int i = 1; i < n; i++) {
			scanf("%d%d%lld", &u, &v, &w);
			mat[u].push_back(make_pair(v, w));
			mat[v].push_back(make_pair(u, w));;
		}
		dfs(1, 0);
		ans = val[1];
		//printf("ans = %d\n", ans);
		ans = ((ans * p[n-1] % mod) * 2) % mod;
		printf("%lld\n", ans % mod);
	}
	return 0;
}