题目大意
求 f(n)=∑ni=0∑ij=02j×j!×S(i,j)
对
998244353
取模
n≤100000
。
题解
S(n,k)=1k!∑i=0k(−1)i(ki)(k−i)n=1k!∑i=0k(−1)ik!i!(k−i)!(k−i)n=∑i=0k(−1)ii!(k−i)n(k−i)!
因为
S(i,j)=0 (i<j)
,所以
f(n)=∑i=0n∑j=0n2j×j!×S(i,j)=∑j=0n2j×j!×∑i=0nS(i,j)=∑j=0n2j×j!×∑i=0n∑l=0j(−1)il!(j−l)i(j−l)!=∑j=0n2j×j!×∑l=0j(−1)il!∑i=0n(j−l)i(j−l)!
设
A(x)=(−1)ii!,B(x)=∑i=0nxix!
所以
B(x)=xn+1−1x!(x−1)
f(n)=∑j=0n2j×j!×∑i=0jA(i)B(j−i)
直接上NTT
时间复杂度: O(nlogn)
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
ll p=998244353;
ll fp(ll a,ll b)
{
ll s=1;
while(b)
{
if(b&1)
s=s*a%p;
a=a*a%p;
b>>=1;
}
return s;
}
namespace ntt
{
ll w1[1000010];
ll w2[1000010];
int rev[1000010];
int n;
void init()
{
n=262144;
int i;
for(i=2;i<=n;i<<=1)
{
w1[i]=fp(3,(p-1)/i);
w2[i]=fp(w1[i],p-2);
}
rev[0]=0;
for(i=1;i<n;i++)
rev[i]=(rev[i>>1]>>1)|(i&1?n>>1:0);
}
void ntt(ll *a,int t)
{
int i,j,k;
ll w,wn,u,v;
for(i=0;i<n;i++)
if(rev[i]<i)
swap(a[i],a[rev[i]]);
for(i=2;i<=n;i<<=1)
{
wn=(t==1?w1[i]:w2[i]);
for(j=0;j<n;j+=i)
{
w=1;
for(k=j;k<j+i/2;k++)
{
u=a[k];
v=a[k+i/2]*w%p;
a[k]=(u+v)%p;
a[k+i/2]=(u-v+p)%p;
w=w*wn%p;
}
}
}
if(t==-1)
{
ll inv=fp(n,p-2);
for(i=0;i<n;i++)
a[i]=a[i]*inv%p;
}
}
};
ll a[500010];
ll b[500010];
ll fac[200010];
int main()
{
ntt::init();
int n;
scanf("%d",&n);
int i;
fac[0]=1;
for(i=1;i<=n;i++)
fac[i]=fac[i-1]*i%p;
a[0]=1;
for(i=1;i<=n;i++)
a[i]=(((i&1?-1:1)*fp(fac[i],p-2))%p+p)%p;
b[0]=1;
b[1]=n+1;
for(i=2;i<=n;i++)
b[i]=((fp(i,n+1)-1)*fp(fac[i]*(i-1)%p,p-2)%p+p)%p;
ntt::ntt(a,1);
ntt::ntt(b,1);
for(i=0;i<ntt::n;i++)
a[i]=a[i]*b[i]%p;
ntt::ntt(a,-1);
ll ans=0;
for(i=0;i<=n;i++)
ans=(ans+fp(2,i)*fac[i]%p*a[i]%p)%p;
printf("%lld\n",ans);
return 0;
}