正題

題目鏈接:https://www.luogu.com.cn/problem/P4091

---

題目大意

給出$n$，求
$$\sum _{i=0}^n\sum _{j=0}^i\left\{\begin{array}{c}i\\ j\end{array}\right\}{2}^{j}j!$$

---

解題思路

看題解才知道$2^{j}j!$對這$n\log n$做法沒有任何意義，卡了好久。
首先斯特林數(shù)的通項公式是$$\left\{\begin{array}{c}n\\ m\end{array}\right\}=\frac{1}{m!}\sum _{k=0}^m(?1{)}^k(\genfrac{}{}{0ex}{}{m}{k})(m?k{)}^n$$
$$?\sum _{k=0}^m\frac{(?1{)}^k(m?k{)}^n}{k!(m?k)!}$$

提到這個式子來（因為如果$j>i$就是$0$所以直接不管這個限制）
$$\sum _{i=0}^n\sum _{j=0}^n{2}^{j}j!\sum _{k=0}^j\frac{(?1{)}^k(j?k{)}^n}{k!(j?k)!}$$
然后把枚舉$i$的那層丟到分數(shù)那里
$$\sum _{j=0}^n{2}^{j}j!\sum _{k=0}^j\frac{(?1{)}^k{\sum }_{i=0}^n(j?k{)}^i}{k!(j?k)!}$$
然后這個后面式子就可以卷積了，定義$F(x)=\frac{(?1{)}^x}{x!},G(x)=\frac{{\sum }_{i=0}^n{x}^n}{x!}$
然后$G$通項公式一下就是$G(x)=\frac{x^{n+1}?1}{(x?1)x!}$
時間復(fù)雜度$O(n\log n)$

---

$code$

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=6e5+10,P=998244353; struct poly{ll a[N],n; }G,F; ll n,ans,fac[N],inv[N],fnv[N],r[N]; ll power(ll x,ll b){ll ans=1;while(b){if(b&1)ans=ans*x%P;x=x*x%P;b>>=1;}return ans; } void NTT(ll *f,ll n,ll op){for(ll i=0;i<n;i++)if(r[i]<i)swap(f[i],f[r[i]]);for(ll p=2;p<=n;p<<=1){ll len=p>>1,tmp=power(3,(P-1)/p);if(op==-1)tmp=power(tmp,P-2);for(ll k=0;k<n;k+=p){ll buf=1;for(ll i=k;i<k+len;i++){ll tt=buf*f[i+len]%P;f[i+len]=(f[i]-tt+P)%P;f[i]=(f[i]+tt)%P;buf=buf*tmp%P;}}}if(op==-1){ll invn=power(n,P-2);for(ll i=0;i<n;i++)f[i]=f[i]*invn%P;}return; } void mul(poly &a,poly &b){ll n=1;while(n<=a.n+b.n)n<<=1;for(ll i=0;i<n;i++)r[i]=(r[i>>1]>>1)^((i&1)?(n>>1):0);NTT(a.a,n,1);NTT(b.a,n,1);for(ll i=0;i<n;i++)a.a[i]=a.a[i]*b.a[i]%P;NTT(a.a,n,-1);return; } int main() {scanf("%lld",&n);fac[1]=fac[0]=fnv[0]=inv[1]=1;for(ll i=2;i<=n;i++)inv[i]=P-(P/i)*inv[P%i]%P;for(ll i=1;i<=n;i++)fac[i]=fac[i-1]*i%P,fnv[i]=inv[i]*fnv[i-1]%P;F.a[0]=G.a[0]=1;F.a[1]=P-1;G.a[1]=n+1;fnv[0]=0;for(ll i=2;i<=n;i++){F.a[i]=(i&1)?(P-fnv[i]):fnv[i];G.a[i]=(power(i,n+1)-1)*inv[i-1]%P*fnv[i]%P;}G.n=F.n=n;mul(G,F);for(ll i=0,pw=1;i<=n;i++){(ans+=G.a[i]*pw%P*fac[i]%P)%=P;pw=pw*2%P;}printf("%lld\n",ans);return 0; }

總結(jié)

以上是生活随笔為你收集整理的P4091-[HEOI2016/TJOI2016]求和【斯特林数,NTT】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。