https://www.luogu.com.cn/problem/P3803
公式:
ω n 1 ≡ g p − 1 n ( m o d p ) \large\omega_n^1\equiv g^{\frac {p-1}n}\pmod p ωn1≡gnp−1(modp)
然后所有单位根运算都可以转成原根了!(前提 p p p 为质数)
p p p 常为 998244353,它的原根 g g g 为 3
实现细节:
- IDFT时 g g g 要取逆元 − g = g − 1 = g p -g=g^{-1}=\frac g p −g=g−1=pg(可以把 g g g 看成单位根然后在单位圆上理解)
- 最后除 n n n 要变成乘上 n n n 的逆元
- 当 p = 99824435 p=99824435 p=99824435 时, p − 1 = 998244352 = 2 23 × 7 × 17 p-1=998244352=2^{23}\times 7\times 17 p−1=998244352=223×7×17,所以可以直接除,此步不用取逆元
#include<bits/stdc++.h>
using namespace std;
#define int long long
inline int read(){int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;
ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+
(x<<3)+(ch^48);ch=getchar();}return x*f;}
#define Z(x) (x)*(x)
#define pb push_back
//#define M
#define mo 998244353
#define N 3000010
#define G 3
int n, m, i, j, k, T;
int a[N], b[N], r[N], l; int pw(int a, int b) {int ans=1; while(b) {if(b&1) ans*=a; b>>=1; a*=a; ans%=mo; a%=mo; }return ans;
}const int invG=pw(G, mo-2); void NTT(int *P, int op) {for(i=0; i<n; ++i) if(i<r[i]) swap(P[i], P[r[i]]); for(i=1; i<n; i<<=1) {int W=pw(op==1 ? G : invG, (mo-1)/(i<<1)); //改动!for(j=0; j<n; j+=(i<<1)) {int w=1; for(k=0; k<i; ++k, w=w*W%mo) { int X=P[j+k], Y=w*P[j+k+i]%mo; P[j+k]=(X+Y)%mo; P[j+k+i]=(X-Y+mo)%mo;}}}
}signed main()
{
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
// srand(time(NULL));
// T=read();
// while(T--) {
//
// }n=read(); m=read(); for(i=0; i<=n; ++i) a[i]=read(); for(i=0; i<=m; ++i) b[i]=read(); m+=n; for(n=1; n<=m; n<<=1) ++l; for(i=0; i<n; ++i) r[i]=((r[i>>1]>>1)|((i&1)<<l-1)); NTT(a, 1); NTT(b, 1); for(i=0; i<n; ++i) a[i]*=b[i], a[i]%=mo; NTT(a, -1); k=pw(n, mo-2); for(i=0; i<=m; ++i) printf("%lld ", a[i]*k%mo); return 0;
}