Luogu P4705 玩游戏
题目描述
Alice 和 Bob 又在玩游戏。
对于一次游戏，首先 Alice 获得一个长度为 的序列 ，Bob 获得一个长度为 的序列 bb。之后他们各从自己的序列里随机取出一个数，分别设为 ，定义这次游戏的 次价值为。
由于他们发现这个游戏实在是太无聊了，所以想让你帮忙计算对于 一次游戏 次价值的期望是多少。
由于答案可能很大，只需要求出模 下的结果即可。
输入输出格式
输入格式：
第一行两个整数 ，分别表示 Alice 和 Bob 序列的长度。
接下来一行 个数，第 个数为 ，表示 Alice 的序列。
接下来一行 个数，第 个数为 ，表示 Bob 的序列。
接下来一行一个整数 ，意义如上所述。
输出格式：
共 行，第 行表示一次游戏 次价值的期望。
输入输出样例
输入样例#1：
复制
1 1
1
2
3
输出样例#1：
复制
3
9
27
输入样例#2：
复制
2 8
764074134 743107904
663532060 183287581 749169979 7678045 393887277 27071620 13482818 125504606
6
输出样例#2：
复制
774481679
588343913
758339354
233707576
36464684
461784746
神仙题啊！！
前置知识：
生成函数。
Taylor展开。
NTT。
多项式求。
次价值得期望就是。
用二项式定理将分子展开:
考虑将展开：
我们设：
答案就是
问题的关键就在于预处理出。
预处理出这个多项式的方法是一个经典套路（根本不会）。
设
这
个
可
以
用
分
治
求
出
。
通过的系数可以预处理出A和B。
解释一下，最后一步变换：
根据Taylor展开：
求导如下：
复合函数求导遵循链式法则：
于是：
我们再处展开，也就是令
得到：
于是我们求出然后就可以用第项得系数来算出和。
代码（算的逆元的时候乘爆了，调了一晚上）：
include<bits/stdc++.h>
define ll long long
define N 300005
define mod 998244353
using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch'-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return xf;}
ll n,m;
int t;
ll a[N],b[N];
ll fac[N<<2],inv[N<<2];
ll ksm(ll t,ll x) {
ll ans=1;
for(;x;x>>=1,t=tt%mod)
if(x&1) ans=ans*t%mod;
return ans;
}
ll Mod(ll a) {
if(a<0) return a+mod;
if(a>=mod) return a-mod;
return a;
}
int rev[N<<2];
void NTT(ll *a,int d,int flag) {
static const ll G=3;
int n=1<<d;
for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<d-1);
for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int s=1;s<=d;s++) {
int len=1<<s,mid=len>>1;
ll w=flag1?ksm(G,(mod-1)/len):ksm(G,mod-1-(mod-1)/len);
for(int i=0;i<n;i+=len) {
ll t=1;
for(int j=0;j<mid;j++,t=tw%mod) {
ll u=a[i+j],v=a[i+j+mid]t%mod;
a[i+j]=Mod(u+v);
a[i+j+mid]=Mod(u-v+mod);
}
}
}
if(flag-1) {
ll inv=ksm(n,mod-2);
for(int i=0;i<n;i++) a[i]=a[i]*inv%mod;
}
}
struct poly {
ll g[N<<2],f[N<<2];
ll a[N<<2],ans[N<<2];
void solve(int l,int r) {
if(lr) return ;
int len=r-l+1,mid=l+r>>1;
solve(l,mid),solve(mid+1,r);
int d=ceil(log2(len+1));
for(int i=0;i<(1<<d);i++) g[i]=f[i]=0;
g[0]=f[0]=1;
for(int i=l;i<=mid;i++) f[i-l+1]=a[i];
for(int i=mid+1;i<=r;i++) g[i-mid]=a[i];
NTT(f,d,1),NTT(g,d,1);
for(int i=0;i<(1<<d);i++) f[i]=f[i]g[i]%mod;
NTT(f,d,-1);
for(int i=l;i<=r;i++) a[i]=f[i-l+1];
}
ll inv[N<<2],A[N<<2];
void Inverse(ll a,int d,ll f) {
f[0]=ksm(a[0],mod-2);
for(int s=1;s<=d;s++) {
int len=1<<s,Len=len<<1;
for(int i=0;i<len;i++) A[i]=a[i],A[i+len]=0;
NTT(A,s+1,1),NTT(f,s+1,1);
for(int i=0;i<Len;i++) f[i]=(2f[i]-f[i]f[i]%modA[i]%mod+mod)%mod;
NTT(f,s+1,-1);
for(int i=len;i<Len;i++) f[i]=0;
}
}
void DER(ll a,int n) {
for(int i=0;i<n;i++) a[i]=a[i+1](i+1)%mod;
a[n]=a[n-1]=0;
}
void INT(ll a,int n) {
for(int i=n-1;i>=0;i--) a[i+1]=a[i]ksm(i+1,mod-2)%mod;
a[0]=0;
}
void Ln(ll *a,int d,ll f) {
Inverse(a,d,inv);
DER(a,1<<d);
NTT(inv,d+1,1),NTT(a,d+1,1);
for(int i=0;i<(1<<d+1);i++) f[i]=inv[i]a[i]%mod;
NTT(f,d+1,-1);
INT(f,1<<d);
}
void mian(int n,int lim,ll *orig,ll f) {
for(int i=1;i<=n;i++) a[i]=orig[i];
solve(1,n);
int d=ceil(log2(lim));
a[0]=1;
Ln(a,d,ans);
for(int i=1;i<=(1<<d);i++) {
f[i]=ans[i];
f[i]=f[i]::inv[i-1]%mod;
if(!(i&1)) f[i]=(mod-f[i])%mod;
}
}
}pre[2];
ll A[N<<2],B[N<<2];
int main() {
n=Get(),m=Get();
for(int i=1;i<=n;i++) a[i]=Get();
for(int i=1;i<=m;i++) b[i]=Get();
t=Get();
fac[0]=1;
for(int i=1;i<=t;i++) fac[i]=fac[i-1]*i%mod;
inv[t]=ksm(fac[t],mod-2);
for(int i=t-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
int lim=max(n,m)+t;
pre[0].mian(n,lim,a,A);
pre[1].mian(m,lim,b,B);
int d=ceil(log2(t*2));
A[0]=n,B[0]=m;
for(int i=t+1;i<(1<<d);i++) A[i]=B[i]=0;
NTT(A,d,1),NTT(B,d,1);
for(int i=0;i<(1<<d);i++) A[i]=A[i]*B[i]%mod;
NTT(A,d,-1);
ll invnm=ksm(1ll*n*m%mod,mod-2);
for(int i=1;i<=t;i++) {
cout<<fac[i]*A[i]%mod*invnm%mod<<"\n";
}
return 0;
}
