题意
题意:给一个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;
}