hdu 6446 - 任意两点路径和

题目链接：点击这里

 

解题思路：

考虑u，v两个点路径对答案的贡献次数，因为是全排列，那么u，v可以放在1，2位置，2.3位置。。。有(n-1)个位置，在颠倒一下顺序就是(n-1)*2,然后其他n-2个点可以随意排，也就是(n-2)!,所以任意一条边的贡献次数就是2*(n-1)!

所以问题就变为求任意两点的路径和，在这个问题上没条边的贡献就是用这条把树节点分为两个部分，那么这个边的贡献次数就是这两部分点的乘积了，这个用dfs求树大小就可以了。

 

#include<cstdio>
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
const int mod = 1e9+7;
int n;
ll dp[maxn],inv[maxn],sum[maxn];
struct Edge
{
    int v, w;
};
vector<Edge> tree[maxn];
void dfs(int cur, int father)
{
    sum[cur] = 1;
    for(int i = 0; i < tree[cur].size(); i++)
    {
        int son = tree[cur][i].v;
        ll len = tree[cur][i].w;
        if(father == son)
            continue;
        dfs(son, cur);
        sum[cur] += sum[son];
        dp[cur] = (dp[cur] + dp[son] + (n-sum[son])*sum[son]%mod*len%mod)%mod;;
    }
}
int main()
{
    int u, v, w, T;
    inv[0] = 1;
    for(int i=1;i<maxn;i++) inv[i] = inv[i-1]*i%mod;
    while(~scanf("%d", &n)){
        for(int i = 1; i <= n; i++)
            tree[i].clear();
        memset(sum, 0, sizeof(sum));
        memset(dp, 0, sizeof(dp));
        for(int i = 0; i < n-1; i++)
        {
            scanf("%d%d%d", &u, &v, &w); 
            Edge t1, t2;
            t1.v = v;
            t1.w = w;
            t2.v = u;
            t2.w = w;
            tree[u].push_back(t1);
            tree[v].push_back(t2);
        }
        dfs(1, -1);
        printf("%lld\n",dp[1]*2%mod*inv[n-1]%mod);
    }
    return 0;
}