解题：HAOI2018 苹果树

题面

统计贡献，每个大小为i的子树贡献就是$i(n-i)$，然后子树里又有$i!$种；同时这个子树的根不确定，再枚举这个根是$r$个放的，又有了$r!$种。子树内选点的方式因为子树的根被钦定了顺序所以只有一个组合数，子树外面的则是一个连乘积。答案就是

$i(n-i)i!r!C_{n-r}^{i-1}\prod\limits_{j=r-1}^{n-i-1}j$

$=(r-1)r*i*i!*(n-i)!$

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 const int N=2005;
 6 int n,mod,ans,fac[N],C[N][N];
 7 void Add(int &x,int y)
 8 {
 9     x+=y;
10     if(x>=mod) x-=mod;
11 }
12 void exGCD(int a,int b,int &x,int &y)
13 {
14     if(!b) x=1,y=0;
15     else exGCD(b,a%b,y,x),y-=a/b*x;
16 }
17 int Inv(int x)
18 {
19     int xx,yy;
20     exGCD(x,mod,xx,yy);
21     return (xx%mod+mod)%mod;
22 }
23 void Pre()
24 {
25     fac[0]=1;
26     for(int i=1;i<=n;i++) fac[i]=1ll*fac[i-1]*i%mod;
27     for(int i=0;i<=n;i++) C[i][0]=1;
28     for(int i=1;i<=n;i++)
29         for(int j=1;j<=i;j++)
30             C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
31 }
32 int main()
33 {
34     scanf("%d%d",&n,&mod),Pre();
35     for(int i=1;i<=n;i++)
36         for(int j=1;j<=n-i+1;j++)
37             Add(ans,(1ll*(i-1)*i*j%mod)*(1ll*fac[j]*fac[n-j]%mod)%mod*C[n-i][j-1]%mod);
38     printf("%d",ans);
39     return 0;
40 }

View Code

 

转载于:https://www.cnblogs.com/ydnhaha/p/10617181.html