[BZOJ5305][组合数学]HAOI2018：苹果树

BZOJ5305

考虑统计点i对其父亲的贡献
枚举i的子树中有j个点，则有$j\ast (n-j)$个点需要通过$fa[i]$
i的子树一共有$j!\ast C_{n-i}^{j-1}$种形态
剩下$n-j-i+1$个点统计进来
$ans={\sum }_{i=1}^n{\sum }_{j=1}^{n-i+1}j!\ast C_{n-i}^{j-1}\ast (n-j)!\ast i\ast (i-1)\ast j$
$O(n^2)$

Code：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=2005;
int c[N][N],fac[N],ans=0;
int n,mod;
int main(){
	n=read(),mod=read();
	fac[0]=1;for(int i=1;i<=n;i++) fac[i]=(ll)i*fac[i-1]%mod;
	for(int i=0;i<=n;i++) c[i][0]=1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++) c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n-i+1;j++)
			ans=(ans+(ll)fac[j]*fac[n-j]%mod*c[n-i][j-1]%mod*i%mod*(i-1)%mod*j%mod)%mod;
	cout<<ans<<endl; 
}