应用背景
思路
易得思路:转为普通多项式+乘法+转回来(X 复杂且慢,浓浓的重工程味)
这里有一个好写且快的思路:
直接转为下降幂点值多项式(利用下降幂单项式系数的EGF易转exp)
发现就是普通方幂多项式(EGF)笛卡尔积一个,就可以得到点值多项式。
反过来,得到点值多项式乘,转回下降幂多项式
所以转为点值,直接点乘后,转回下降幂多项式即可。
PS.看到网上说这个下降幂多项式转点值egf叫FDT(反过来IFDT)。
code:
点击查看代码
#include<bits/stdc++.h>
using namespace std;
typedef double db;
typedef long long ll;
const int N=1e6+5;
const ll mod=998244353;
ll inv[N],inv3,f[N],gen[2][N];
ll ksm(ll a,ll b) {ll mul=1;for(;b;b>>=1,a=a*a%mod)if(b&1)mul=mul*a%mod;return mul;}
int rev[N],L,up;
void NTT(ll *a,int op) {
for(int i=0;i<up;i++) {
if(rev[i]>i)swap(a[i],a[rev[i]]);
}
for(int mid=1;mid<up;mid<<=1) {
int len=mid<<1;ll w1=gen[op][len];
for(int l=0;l<up;l+=len) {
ll W=1;
for(int i=0;i<mid;i++,W=W*w1%mod) {
int p=l+i,q=p+mid;
ll x=a[p],y=W*a[q];
a[p]=(x+y)%mod;a[q]=(x-y)%mod;
}
}
}
}
void gt_up(int len) {
up=1,L=0;
while(up<len) {up<<=1,L++;}
for(int i=1;i<up;i++) {rev[i]=(rev[i>>1]>>1)|((i&1)<<(L-1));}
}
void gt_gen(int len) {
inv3=ksm(3,mod-2);
gt_up(len);
gen[0][up]=ksm(3,(mod-1)/up);gen[1][up]=ksm(inv3,(mod-1)/up);
for(int i=up;i;i>>=1) {
gen[0][i>>1]=gen[0][i]*gen[0][i]%mod;
gen[1][i>>1]=gen[1][i]*gen[1][i]%mod;
}
}
ll jc[N],ijc[N];
void init(int n) {
jc[0]=1;for(int i=1;i<=n;i++)jc[i]=jc[i-1]*i%mod;
ijc[n]=ksm(jc[n],mod-2);for(int i=n;i;i--)ijc[i-1]=ijc[i]*i%mod;
}
ll e[N];
void FDT(int n,ll *f,int op) {
for(int i=0;i<n;i++) {
if(op!=1&&(i&1)) {e[i]=-ijc[i];}
else e[i]=ijc[i];
}
for(int i=n;i<up;i++)e[i]=f[i]=0;
NTT(e,1),NTT(f,1);
for(int i=0;i<up;i++)f[i]=f[i]*e[i]%mod;
NTT(f,0);
for(int i=0,j=ksm(up,mod-2);i<n;i++)f[i]=f[i]*j%mod;
for(int i=n;i<up;i++)f[i]=0;
}
ll F[N],G[N];
int main() {
int n,m;scanf("%d%d",&n,&m);
n++;m++;
int lim=n+m-1;
init(lim);gt_gen(lim<<1);
for(int i=0;i<n;i++)scanf("%lld",&F[i]);
for(int i=0;i<m;i++)scanf("%lld",&G[i]);
FDT(lim,F,1),FDT(lim,G,1);
for(int i=0;i<lim;i++)F[i]=F[i]*G[i]%mod*jc[i]%mod;
FDT(lim,F,0);
for(int i=0;i<n+m-1;i++) {printf("%lld ",(F[i]+mod)%mod);}
return 0;
}
</details>
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人