Tree and Permutation

题意
题意：给一个n（1e5）个点，n-1条边的带权树，按结点进行全排列，对每个全排列，求其每两个相邻数字所代表的结点之间的权值之和sum1。然后再然后再将所有排列的sum1加起来再摸1e9+7，比如排列1423，就是先求树中结点1和结点4的距离（权值和）。然后再加上结点4和结点2的距离，最后加上结点2和结点3的距离。

初步分析
我们随便取该树中的一条边i.如果i连接的右边有x个结点。左边有n-x个结点.那么有x*(n-x)种组合。没一种组合代表相邻的两个数。这两个数一定经过这一条边。那么有多少种排列会经过边i呢。首先这两个数可以互换顺序比如12,21.然后其他n-2个数可以排列成(n-2)!种可能。且这两个数可以在排列种有(n-1)种位置。所以经过边i的排列就一共有2 * (n-x) * x * (n-1)! * w种.所以我们把每条边都计算出来求和就行了。那么怎么计算x呢?x就是i边连接的子节点为根的子树的结点数可以通过树状DP来求。

直接上代码

#include <iostream>
using namespace std;
const int MAX_N=1e5+10;
const int MAX_M=2*MAX_N;
const int mod=1e9+7;

int ans=0,n,result=0;
int head[MAX_N];
int jiechen[MAX_N];
int cnt[MAX_N];
struct edge{
    int v;
    int next;
    int w;
}eid[MAX_M];

void insert(int u,int v,int w){
    eid[ans].v=v;
    eid[ans].w=w;
    eid[ans].next=head[u];
    head[u]=ans++;
}

void dfs(int node,int father){
    cnt[node]=1;
    for(int i=head[node];i!=-1;i=eid[i].next){
        int v=eid[i].v;
        if(v==father) continue;
        dfs(v,node);
        cnt[node]+=cnt[v];
    }
}

void find_jiechen(int i,int n){
    if(i==1)
        jiechen[i]=1;
    else if(i<=n){
        jiechen[i]=i*jiechen[i-1];
    }
}

void sum(int node,int father){
    for(int i=head[node];i!=-1;i=eid[i].next){
        int v=eid[i].v;
        if(v==father) continue;
        result+=(2*cnt[v]*(n-cnt[v])*eid[i].w%mod)*(jiechen[n-1]%mod)%mod;//ab mod c=(a mod c)(b mod c)mod c 
    }
}

int main() {
    cin>>n;
    int u,v,w;
    find_jiechen(1,n);

    for(int i=1;i<n;++i){
        scanf("%d %d %d",&u,&v,&w);
        insert(u,v,w);
        insert(v,u,w);
    }

    cout<<result;

    return 0;
}