bzoj 5305 [Haoi2018]苹果树

https://www.lydsy.com/JudgeOnline/problem.php?id=5305
http://www.elijahqi.win/archives/3680
考虑每条边对答案的贡献 设i节点的子树大小为$size_i$那么对于答案的贡献就是$n*(n-size_i)$
发现如果不考虑生成树的顺序那么生成树的方案是节点的阶乘种方案
那么可以考虑枚举每个节点 然后再枚举他子树大小是多少来计算答案这样算出方案 然后再*对答案的贡献即可
那么子树的方案数自然就是$size_i!*C_{n-i}^{size_i-1}$ 因为需要保证子树内的节点编号比i大
子树外生成方式自然是
首先生成i 有i!种方式
$(i+1-2)\times (i+2-2)\times... (n-size_i+1 -2 )$
相当于仍然用同样的方法考虑只不过这时不可以考虑i节点下放的放置方案
那么$i!\times (i+1-2)\times (i+2-2)\times... (n-size_i+1 -2 )$
可以化简$i\times (i-1)\times (n-size_i-1)!$
然后$n^2$计算即可

#include<cstdio>
#include<cctype>
#include<algorithm>
#define ll long long
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if(T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}
const int N=2200;
int c[N][N],n,mod,jc[N],ans;
inline int inc(int &x,int v){return x+v>=mod?x+v-mod:x+v;}
int main(){
    freopen("bzoj5305.in","r",stdin);
    n=read();mod=read();jc[0]=1;
    for (int i=0;i<=n;++i) c[i][0]=1;
    for (int i=1;i<=n;++i)
        for (int j=1;j<=i;++j) c[i][j]=inc(c[i-1][j-1],c[i-1][j]);
    for (int i=1;i<=n;++i) jc[i]=(ll)jc[i-1]*i%mod;
    for (int i=2;i<=n;++i){
        for (int sz=1;sz<=n-i+1;++sz){
            ans=inc(ans,(ll)sz*(n-sz)*i*(i-1)%mod*c[n-i][sz-1]%mod*jc[sz]%mod*jc[n-sz-1]%mod);
        }
    }printf("%d\n",ans);
    return 0;
}