5.13 省选模拟赛 优雅的绽放吧,墨染樱花 多项式 prufer序列 计数 dp
作者：互联网
LINK：优雅的绽放吧,墨染樱花
当时考完只会50分的做法 最近做了某道题受到启发 故会做这道题目了。(末尾附30分 50分 100分code
看到度数容易想到prufer序列 考虑dp统计方案数。
设f[i][j]表示前i个数字占了prufer序列j个位置的方案数.最后答案为f[n][n-2].
容易想到转移 \(f[i][j]+=f[i-1][k]\cdot C(n-k,j-k)\cdot w_i^{j-k+1}\cdot (j-k+1)\)
复杂度n^3 期望得分30.
容易发现第二维是一个卷积 NTT优化 复杂度n^2log 期望得分50.
考虑满分做法：若知道一个度数排列 满足\(\sum_i d_i=n-2\)
那么容易计算出贡献. 贡献为\(\frac{(n-2)!}{\Pi (d_i!)}\Pi (d_i+1)w_i^{d_i+1}\)
考虑对于这个式子先进行化简:\((n-2)!\Pi w_i\frac{1}{\Pi (d_i!)}\Pi (d_i+1)w_i^{d_i}\)
那么前面是一个常数 考虑对后面的式子进行变形.
由于存在阶乘 考虑设出EGF来解决问题。
对于第i个点 设F(x)为其贡献的EGF. \(F(x)=\sum_{j=0}^{\infty}\frac{w_i^{j}x^j(j+1)}{j!}\)
那么我们要求的就是 \([x^{n-2}]\Pi F(x_i)\)
直接求复杂度还是n^2log 这个连乘很麻烦 考虑化简.
考虑取对数 然后再exp回来 逐个取复杂度还是很高。
一个trick 设\(G(x)=\sum_{j=0}^{\infty}\frac{x^j(j+1)}{j!}\)
那么对G(x)取对数后对应项乘以 \(w_i^k\) 那么和刚才的F(x)取对数等价。
关键如何求\(\sum_{k=0}\sum_{i=1}^n w_i^k\)
还是设出上式的生成函数W(x) 容易得到\(W(x)=\sum_{i=1}^n\frac{1}{1-w_ix}\)
\(W(x)=\sum_{i=1}^n1-\frac{w_ix}{1-w_ix}\)
一个trick \([Ln(1-w_ix)]'=-\frac{w_i}{1-w_ix}\)
此时有 \(W(x)=n-x\sum_{i=1}^n(Ln(1-w_ix))\)
那么存在\(W(x)=n-x(Ln(\Pi_{}^n(1-w_ix)))'\)
里面那个东西可以直接分治NTT求.
综上分为两部 前者nlogn 后者 nlog^2. 期望得分100.
score 30
const int MAXN=5010;
int n;
int w[MAXN],fac[MAXN],inv[MAXN];
int f[MAXN][MAXN];//f[i][j]表示前i个数填了j个位置的贡献.
inline int ksm(int b,int p)
{
int cnt=1;
while(p){if(p&1)cnt=(ll)cnt*b%mod;p=p>>1;b=(ll)b*b%mod;}
return cnt;
}
inline int C(int a,int b){if(a<b)return 0;return (ll)fac[a]*inv[b]%mod*inv[a-b]%mod;}
int main()
{
freopen("yuyuko.in","r",stdin);
freopen("yuyuko.out","w",stdout);
get(n);fac[0]=1;
rep(1,n,i)get(w[i]),fac[i]=(ll)fac[i-1]*i%mod;
inv[n]=ksm(fac[n],mod-2);
fep(n-1,0,i)inv[i]=(ll)inv[i+1]*(i+1)%mod;
f[0][0]=1;
rep(1,n,i)
{
rep(0,n-2,j)
{
int ww=1;
rep(1,j+1,k)
{
ww=(ll)ww*w[i]%mod;
f[i][j]=(f[i][j]+(ll)f[i-1][j-k+1]*C(n-2-j+k-1,k-1)%mod*k%mod*ww)%mod;
}
}
}
put(f[n][n-2]);
return 0;
}
score 50
const int MAXN=200010,G=3;
int n,lim,INV;
int w[MAXN],rev[MAXN],fac[MAXN],inv[MAXN];
int f[MAXN],g[MAXN],c[MAXN];//f[i][j]表示前i个数填了j个位置的贡献.
inline int ksm(int b,int p)
{
int cnt=1;
while(p){if(p&1)cnt=(ll)cnt*b%mod;p=p>>1;b=(ll)b*b%mod;}
return cnt;
}
inline int C(int a,int b){if(a<b)return 0;return (ll)fac[a]*inv[b]%mod*inv[a-b]%mod;}
inline void NTT(int *a,int op)
{
rep(1,lim-1,i)if(i<rev[i])swap(a[i],a[rev[i]]);
for(int len=2;len<=lim;len=len<<1)
{
int mid=len>>1;
int wn=ksm(G,op==1?(mod-1)/len:mod-1-(mod-1)/len);
for(int j=0;j<lim;j+=len)
{
int d=1;
for(int i=0;i<mid;++i)
{
int x=a[i+j],y=(ll)a[i+j+mid]*d%mod;
a[i+j]=(x+y)%mod;a[i+j+mid]=(x-y+mod)%mod;
d=(ll)d*wn%mod;
}
}
}
}
int main()
{
freopen("yuyuko.in","r",stdin);
freopen("yuyuko.out","w",stdout);
get(n);fac[0]=1;f[0]=1;
rep(1,n,i)get(w[i]),fac[i]=(ll)fac[i-1]*i%mod;
inv[n]=ksm(fac[n],mod-2);
fep(n-1,0,i)inv[i]=(ll)inv[i+1]*(i+1)%mod;
lim=1;while(lim<n-2+n-2+1)lim=lim<<1;
rep(1,lim-1,i)rev[i]=rev[i>>1]>>1|((i&1)?lim>>1:0);
INV=ksm(lim,mod-2);
rep(1,n,i)
{
int ww=1;
rep(0,n-2,j)
{
ww=(ll)ww*w[i]%mod;
f[j]=(ll)f[j]*fac[n-2-j]%mod;
g[j]=(ll)inv[j]*(j+1)%mod*ww%mod;
}
rep(n-1,lim-1,j)f[j]=0,g[j]=0;
NTT(g,1);NTT(f,1);rep(0,lim-1,j)f[j]=(ll)f[j]*g[j]%mod;
NTT(f,-1);rep(0,n-2,j)f[j]=(ll)f[j]*INV%mod*inv[n-2-j]%mod;
}
put(f[n-2]);
return 0;
}
score 100
const int MAXN=300010,GG=3;
int n;
int w[MAXN],O[MAXN],rev[MAXN],f[MAXN],g[MAXN];
int fac[MAXN],inv[MAXN],C[MAXN],D[MAXN],ans[MAXN];
int A[20][MAXN],tv[MAXN],v[MAXN],E[MAXN],F[MAXN],G[MAXN];
inline int ksm(int b,int p)
{
int cnt=1;
while(p)
{
if(p&1)cnt=(ll)cnt*b%mod;
b=(ll)b*b%mod;p=p>>1;
}
return cnt;
}
inline void NTT(int *a,int op,int lim)
{
rep(1,lim-1,i)
{
rev[i]=rev[i>>1]>>1|((i&1)?lim>>1:0);
if(i<rev[i])swap(a[i],a[rev[i]]);
}
for(int len=2;len<=lim;len=len<<1)
{
int mid=len>>1;
int wn=ksm(GG,op==1?(mod-1)/len:mod-1-(mod-1)/len);
rep(1,mid-1,i)O[i]=(ll)O[i-1]*wn%mod;
for(int j=0;j<lim;j+=len)
{
for(int i=0;i<mid;++i)
{
int x=a[i+j],y=(ll)a[i+j+mid]*O[i]%mod;
a[i+j]=(x+y)%mod;a[i+j+mid]=(x-y+mod)%mod;
}
}
}
if(op==-1)for(int i=0,INV=ksm(lim,mod-2);i<=lim-1;++i)a[i]=(ll)a[i]*INV%mod;
}
inline void solve(int l,int r,int d)
{
if(l==r)return A[d][0]=1,A[d][1]=mod-w[l],void();
int mid=(l+r)>>1;
solve(l,mid,d);solve(mid+1,r,d+1);
int lim=1;while(lim<=r-l+1)lim=lim<<1;
rep(mid-l+2,lim-1,i)A[d][i]=0;
rep(r-mid+1,lim-1,i)A[d+1][i]=0;
NTT(A[d],1,lim);NTT(A[d+1],1,lim);
rep(0,lim-1,i)A[d][i]=(ll)A[d][i]*A[d+1][i]%mod;
NTT(A[d],-1,lim);
}
inline void Direv(int *A,int *B,int len)
{
rep(1,len-1,i)B[i-1]=(ll)A[i]*i%mod;B[len-1]=0;
}
inline void Inv(int *A,int *B,int len)
{
if(len==1)return B[0]=ksm(A[0],mod-2),void();
Inv(A,B,len>>1);rep(0,len-1,i)E[i]=A[i],F[i]=B[i];
NTT(E,1,len<<1);NTT(F,1,len<<1);
rep(0,(len<<1)-1,i)E[i]=(ll)E[i]*F[i]%mod*F[i]%mod;
NTT(E,-1,len<<1);
rep(0,len-1,i)B[i]=((ll)2*B[i]-E[i]+mod)%mod;
rep(0,(len<<1)-1,i)E[i]=F[i]=0;
}
inline void Inter(int *A,int *B,int len)
{
rep(1,len-1,i)B[i]=(ll)A[i-1]*inv[i]%mod*fac[i-1]%mod;B[0]=0;
}
inline void Ln(int *A,int *B,int len)
{
Direv(A,C,len);Inv(A,D,len);
NTT(C,1,len<<1);NTT(D,1,len<<1);
rep(0,(len<<1)-1,i)C[i]=(ll)C[i]*D[i]%mod;
NTT(C,-1,len<<1);Inter(C,B,len);
rep(0,(len<<1)-1,i)C[i]=D[i]=0;
}
inline void Exp(int *A,int *B,int len)
{
if(len==1)return B[0]=1,void();
Exp(A,B,len>>1);Ln(B,G,len);
G[0]=(A[0]+1-G[0]+mod)%mod;
rep(1,len-1,i)G[i]=(A[i]-G[i]+mod)%mod;
NTT(B,1,len<<1);NTT(G,1,len<<1);
rep(0,(len<<1)-1,i)B[i]=(ll)B[i]*G[i]%mod;
NTT(B,-1,len<<1);
rep(len,(len<<1)-1,i)B[i]=G[i]=0;
}
int main()
{
freopen("yuyuko.in","r",stdin);
freopen("yuyuko.out","w",stdout);
get(n);
if(n==1){puts("0");return 0;}
rep(1,n,i)get(w[i]);
//先求出G(x)=1+ai^kx^k;
int len=1;fac[0]=1;O[0]=1;
while(len<=n)len=len<<1;
rep(1,(len<<1),i)fac[i]=(ll)fac[i-1]*i%mod;
inv[len<<1]=ksm(fac[len<<1],mod-2);
fep((len<<1)-1,0,i)inv[i]=(ll)inv[i+1]*(i+1)%mod;
solve(1,n,0);rep(0,n,i)tv[i]=A[0][i];
Ln(tv,v,len);v[0]=n;
rep(1,len-1,i)v[i]=mod-(ll)v[i]*i%mod;
rep(0,len-1,i)f[i]=(ll)inv[i]*(i+1)%mod;
Ln(f,g,len);
rep(0,len-1,i)g[i]=(ll)g[i]*v[i]%mod;
Exp(g,ans,len);
int res=(ll)ans[n-2]*fac[n-2]%mod;
rep(1,n,i)res=(ll)res*w[i]%mod;
put(res);return 0;
}
标签：cnt,5.13,省选,墨染,ll,int,MAXN,rep,mod 来源： https://www.cnblogs.com/chdy/p/12960547.html