2018CCPC Tree and Permutation

题目

题意见其他博客（滑稽）

首先，推公式，有N!种全排列，每种排列有N-1条路，总共有N!*(N-1)条路，自己在纸上写一下，可以发现，每种路径（x-y）出现两次，（x-y和y-x是一样的长度），N个点，有N*(N-1)/2不同的种组合，计算出每种组合出现2*（N-1）！次。

本题即树上搜索和DP，计算每两点之间的距离之和，再乘以出现的次数即可。

注意点：注意取模操作，千万小心爆long long

树上距离之和计算方法：计算每一条边的对答案的贡献，即 经过的次数*边的权重 经过的次数的计算方法，即一条边的两端的子树的节点数相乘，即左边选一个点到右边其中一个点的方法。一遍DFS即可。

#include<iostream>
#include<stdio.h>
#define maxn 300005
#include<string.h>
#define ll long long
using namespace std;
const ll MOD=1e9+7;
int n;
struct edge{
	int next;
	int to;
	ll w;
};

edge eg[maxn];

int head[maxn];
int cnt=0;
void Add(int u,int v,ll w)
{
	eg[++cnt].next=head[u];
	eg[cnt].w=w;
	eg[cnt].to=v;
	head[u]=cnt;
}

ll sum[maxn],dp[maxn];
void dfs(int cur,int fa)
{
	sum[cur]=1;
	for(int i=head[cur];~i;i=eg[i].next)
	{
		int son=eg[i].to;
		ll len=eg[i].w;
		if(fa==son) continue;//因为无向图，所以防止死循环 
		dfs(son,cur);
		sum[cur]+=sum[son];//将孩子的节点数信息反馈给父节点 
		dp[cur]+=(dp[son]%MOD+((n-sum[son])%MOD)*(sum[son]*len)%MOD)%MOD;
		
	}
}


ll p[100006];
int main()
{
	ll ans,c;
	int N,a,b;
	p[0]=1;
	for(int i=1;i<100005;i++)
	{
		p[i]=(p[i-1]%MOD)*(i%MOD)%MOD;//写成这样防止爆ll 
	}
	while(scanf("%d",&N)==1)
	{
		cnt=0;
		ans=0;
		memset(dp,0,sizeof(dp));
		memset(sum,0,sizeof(sum));
		memset(eg,0,sizeof(eg));
		n=N;
		memset(head,-1,sizeof(head));
		for(int k=1;k<N;k++)
		{
			scanf("%d%d%lld",&a,&b,&c);
			Add(a,b,c);
			Add(b,a,c);
		}
		dfs(1,-1);
		ans=(2*dp[1]%MOD)*p[n-1];
		printf("%lld\n",ans%MOD);
	}
}