[题解] LuoguP4091 [HEOI2016/TJOI2016]求和
生活随笔
收集整理的這篇文章主要介紹了
[题解] LuoguP4091 [HEOI2016/TJOI2016]求和
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
傳送門
首先我們來看一下怎么求\(S(m,n)\)。
注意到第二類斯特林數的組合意義就是將\(m\)個不同的物品放到\(n\)個沒有區別的盒子里，不允許有空盒子的方案數。
那么將\(m\)個不同的物品隨便扔到\(n\)個盒子里的方案數就是\(n^m\)，這里盒子也有區別了。
那么枚舉有多少盒子有物品，然后斯特林數安排一下，注意到這是的盒子是沒有區別的，再排列就好了，即
\[n^m=\sum\limits_{i=0}^n \binom{n}{i}S(m,i)i!
\]
\]
但我們要求的是\(S\)，這里又有組合數，直接二項式反演，，得到
\[n! \times S(m,n)=\sum\limits_{i=0}^n (-1)^{n-i}\binom{n}{i}i^m
\]
\]
所以
\[S(m,n) = \frac{1}{n!}\sum\limits_{i=0}^n (-1)^{n-i} \binom{n}{i} i^m
\]
\]
然后再來看題目里的柿子：
\[\sum\limits_{i=0}^n\sum\limits_{j=0}^i S(i,j) \times 2^j \times (j!)
\]
\]
我也不知道它為什么長成這樣...
由于\(j>i\)時\(S(i,j)=0\)，所以\(j\)完全可以到\(n\)，然后再吧\(S(i,j)\)換成上面的求和形式
\[\sum\limits_{i=0}^n\sum\limits_{j=0}^n 2^j \sum\limits_{k=0}^j (-1)^{j-k}\binom{j}{k} k^i
\]
\]
\[= \sum\limits_{j=0}^n 2^j \sum\limits_{k=0}^j (-1)^{j-k}\binom{j}{k} \sum\limits_{i=0}^n k^i
\]
\]
\[= \sum\limits_{j=0}^n 2^j \sum\limits_{k=0}^j (-1)^{j-k}\frac{j!}{k!(j-k)!} \sum\limits_{i=0}^n k^i
\]
\]
\[= \sum\limits_{j=0}^n 2^j\times (j!) \sum\limits_{k=0}^j \frac{(-1)^{j-k}}{(j-k)!} \times (k!\sum\limits_{i=0}^n k^i)
\]
\]
后面已經是一個卷積的形式了，具體的，令
\[f_i = \frac{(-1)^i}{i!}
\]
\]
\[g_i = i!\sum\limits_{k=0}^n i^k = i! \times \frac{(i^{k+1}-1)}{i-1}
\]
\]
需要特判\(g_1=n+1\)，把\(f,g\)卷起來求和就好了
\(Code:\)
#include <bits/stdc++.h>
using namespace std;
const int N=3e5+10,P=998244353;
inline int fpow(int x,int y,int mod=P)
{
int ret=1; for(x%=mod;y;y>>=1,x=1ll*x*x%mod)
if(y&1) ret=1ll*ret*x%P;
return ret;
}
const int gen=3,igen=fpow(gen,P-2);
inline int add(int x,int y,int mod=P){return (x+=y)>=mod?x-mod:x;}
inline int sub(int x,int y,int mod=P){return (x-=y)<0?x+mod:x;}
inline int normal(int x,int mod=P){return x<0?x+mod:x;}
namespace Poly
{
int rev[N];
void init(int n)
{
for(int i=0;i<n;i++)
rev[i]=rev[i>>1]>>1|((i&1)?n>>1:0);
}
void ntt(int *f,int n,int flg)
{
for(int i=0;i<n;i++)
if(rev[i]<i) swap(f[i],f[rev[i]]);
for(int len=2,k=1;len<=n;len<<=1,k<<=1)
{
int wn=fpow(flg==1?gen:igen,(P-1)/len);
for(int i=0;i<n;i+=len)
for(int j=i,w=1;j<i+k;j++,w=1ll*w*wn%P)
{
int tmp=1ll*f[j+k]*w%P;
f[j+k]=sub(f[j],tmp),f[j]=add(f[j],tmp);
}
}
if(flg==-1)
{
int inv=fpow(n,P-2);
for(int i=0;i<n;i++) f[i]=1ll*f[i]*inv%P;
}
}
}
using Poly::ntt;
int f[N],g[N],n;
int pw2[N],inv[N],ifac[N],fac[N];
int main()
{
scanf("%d",&n);
inv[1]=ifac[0]=ifac[1]=1;
pw2[0]=1,pw2[1]=2; fac[0]=fac[1]=1;
for(int i=2;i<=n;i++)
{
inv[i]=1ll*inv[P%i]*(P-P/i)%P;
ifac[i]=1ll*ifac[i-1]*inv[i]%P;
pw2[i]=2ll*pw2[i-1]%P;
fac[i]=1ll*fac[i-1]*i%P;
}
for(int i=0;i<=n;i++)
{
f[i]=1ll*((i&1)?P-1:1)*ifac[i]%P;
if(i==1) g[i]=n+1;
else g[i]=1ll*sub(fpow(i,n+1),1)*ifac[i]%P*fpow(i-1,P-2)%P;
}
int limit=1; while(limit<=n*2)limit<<=1;
Poly::init(limit);
ntt(f,limit,1),ntt(g,limit,1);
for(int i=0;i<limit;i++) f[i]=1ll*f[i]*g[i]%P;
ntt(f,limit,-1);
int ans=0;
for(int i=0;i<=n;i++) ans=add(ans,1ll*pw2[i]*fac[i]%P*f[i]%P);
printf("%d\n",ans);
return 0;
}
總結
以上是生活随笔為你收集整理的[题解] LuoguP4091 [HEOI2016/TJOI2016]求和的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： Vision Transformer 论
- 下一篇： pytorch必须掌握的的4种学习率衰减