多项式快速幂

模板

题意:给多项式 f ( x ) f(x) f(x) 和正整数 k k k,求 f ( x ) k ( m o d x n ) f(x)^k\pmod{x^n} f(x)k(modxn),系数对 998244353 998244353 998244353 取模。
k ≤ 1 0 1 0 5 , a 0 = 1 k\le10^{10^5},a_0=1 k10105,a0=1

前置知识:多项式 ln ⁡ \ln ln,多项式 exp ⁡ \exp exp

下面多项式的系数默认对 998244353 998244353 998244353 取模。

对于求多项式的快速幂,很容易想到整数的快速幂,用类似的方法求出。

但是对于此题, k ≤ 1 0 1 0 5 k\le10^{10^5} k10105,太大了,会 TLE。

想到之前学过的多项式 ln ⁡ \ln ln,多项式 exp ⁡ \exp exp,对原式取 ln ⁡ \ln ln 再取 exp ⁡ \exp exp 可得
f ( x ) k ≡ e k ln ⁡ f ( x ) ( m o d x n ) f(x)^k\equiv e^{k\ln f(x)}\pmod{x^n} f(x)keklnf(x)(modxn)

这样先做一遍多项式 ln ⁡ \ln ln,再做一遍多项式 exp ⁡ \exp exp,就可以求出答案了。

但是这里还有个问题, k k k 那么大,如果对他取模,正确性能否保证呢?如果不行,那怎么办呢?

这里有一个结论: f ( x ) p ≡ f ( x p ) f(x)^p\equiv f(x^p) f(x)pf(xp)

证明:使用数学归纳法。如果 f ( x ) f(x) f(x) 只有常数项,显然上式成立。
f ( x ) f(x) f(x) 的最高次为 t t t g ( x ) g(x) g(x) f ( x ) f(x) f(x) 去掉最高次项的多项式,且满足 g ( x ) p ≡ g ( x p ) g(x)^p\equiv g(x^p) g(x)pg(xp)
f ( x ) = g ( x ) + a t x t f(x)=g(x)+a_tx^t f(x)=g(x)+atxt,得出 f ( x ) p = ( g ( x ) + a t x t ) p f(x)^p=(g(x)+a_tx^t)^p f(x)p=(g(x)+atxt)p
有一个结论 ( a + b ) p ≡ a p + b p ( m o d p ) (a+b)^p\equiv a^p+b^p\pmod p (a+b)pap+bp(modp),证明略。
得到 f ( x ) p ≡ g ( x ) p + a t p ( x p ) t f(x)^p\equiv g(x)^p+a_t^p(x^p)^t f(x)pg(x)p+atp(xp)t
根据费马小定理, a p ≡ a ( m o d p ) a^p\equiv a\pmod p apa(modp),得 f ( x ) p ≡ g ( x p ) + a t ( x p ) t f(x)^p\equiv g(x^p)+a_t(x^p)^t f(x)pg(xp)+at(xp)t,得证。

对于本题, n ≤ 1 0 5 , p = 998244353 n\le10^5,p=998244353 n105,p=998244353,所以 n < p n

n<p,那么 f ( x ) p ≡ f ( x p ) ≡ a 0 ≡ 1 ( m o d x n ) f(x)^p\equiv f(x^p)\equiv a_0\equiv1\pmod{x^n} f(x)pf(xp)a01(modxn)

所以 f ( x ) k ≡ f ( x ) k   m o d   p ( m o d x n ) f(x)^k\equiv f(x)^{k\bmod p}\pmod{x^n} f(x)kf(x)kmodp(modxn)

又可以得到 f ( x ) k   m o d   p ≡ e ( k   m o d   p ) ln ⁡ f ( x ) ( m o d x n ) f(x)^{k\bmod p}\equiv e^{(k\bmod p)\ln f(x)}\pmod{x^n} f(x)kmodpe(kmodp)lnf(x)(modxn)

这样就可做了。

时间复杂度为 O ( n log ⁡ n ) O(n\log n) O(nlogn)

代码

#include
using namespace std;
typedef long long ll;
const int N=(1<<18)+1;
const ll mod=998244353,g=3,inv2=499122177;
int len=1,n,k;
ll a1[N],w,wn,a[N],ans[N],invans[N],lnans[N],da[N],inva[N],a2[N];
ll ksm(ll a,ll b)
{
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void change(ll num[])
{
    for(int i=1,j=len/2;i<len-1;i++){
        if(i<j) swap(num[i],num[j]);
        int k=len/2;
        while(j>=k) j-=k,k>>=1;
        if(j<k) j+=k;
    }
}
void ntt(ll num[],int fl)
{
    for(int i=2;i<=len;i<<=1){
        if(fl==1) wn=ksm(g,(mod-1)/i);
        else wn=ksm(g,mod-1-(mod-1)/i);
        for(int j=0;j<len;j+=i){
            w=1;
            for(int k=j;k<j+i/2;k++){
                ll u=w*num[k+i/2]%mod,t=num[k];
                num[k]=(t+u)%mod;
                num[k+i/2]=(t-u+mod)%mod;
                w=w*wn%mod;
            }
        }
    }
    if(fl==-1){
        ll inv=ksm(len,mod-2);
        for(int i=0;i<len;i++) num[i]=num[i]*inv%mod;
    }
}
void getinv(int n,ll a[],ll ans[])
{
	if(n==1){ans[0]=ksm(a[0],mod-2);return;}
	getinv((n+1)/2,a,ans);
	len=1;
	while(len<2*n) len*=2;
	for(int i=0;i<n;i++) a1[i]=a[i];
	for(int i=n;i<len;i++) a1[i]=0;
	change(a1),change(ans);
	ntt(a1,1),ntt(ans,1);
	for(int i=0;i<len;i++) ans[i]=ans[i]*(2-ans[i]*a1[i]%mod+mod)%mod;
	change(ans),ntt(ans,-1);
	for(int i=n;i<len;i++) ans[i]=0;
}
void getln(int n,ll a[],ll ln[])
{
    memset(da,0,sizeof(da));
	for(int i=1;i<n;i++) da[i-1]=a[i]*i%mod;
    da[n-1]=0;
    memset(inva,0,sizeof(inva));
	getinv(n,a,inva);
	len=1;
	while(len<2*n) len*=2;
	change(da),change(inva);
	ntt(da,1),ntt(inva,1);
	for(int i=0;i<len;i++) ln[i]=da[i]*inva[i]%mod;
	change(ln),ntt(ln,-1);
	for(int i=len-1;i>=0;i--) ln[i+1]=ksm(i+1,mod-2)*ln[i]%mod;
    for(int i=n;i<len;i++) ln[i]=0;
	ln[0]=0;
}
void getsqrt(int n,ll a[],ll ans[])
{
    if(n==1){ans[0]=a[0];return;}
    getsqrt((n+1)/2,a,ans);
    len=1;
    while(len<2*n) len*=2;
    memset(invans,0,sizeof(invans));
    getinv(n,ans,invans);
    for(int i=0;i<n;i++) a1[i]=a[i];
    for(int i=n;i<len;i++) a1[i]=0;
    change(a1),change(invans);
    ntt(a1,1),ntt(invans,1);
    for(int i=0;i<len;i++) a1[i]=a1[i]*invans[i]%mod;
    change(a1),ntt(a1,-1);
    for(int i=0;i<n;i++) ans[i]=(a1[i]+ans[i])*inv2%mod;
    for(int i=n;i<len;i++) ans[i]=0;
}
void getexp(int n,ll a[],ll ans[])
{
    if(n==1){ans[0]=1;return;}
    getexp((n+1)/2,a,ans);
    len=1;
    while(len<2*n) len*=2;
    getln(n,ans,lnans);
    for(int i=0;i<n;i++) lnans[i]=(-lnans[i]+a[i]+mod)%mod;
    lnans[0]++;
    change(ans),change(lnans);
    ntt(ans,1),ntt(lnans,1);
    for(int i=0;i<len;i++) ans[i]=ans[i]*lnans[i]%mod;
    change(ans),ntt(ans,-1);
    for(int i=n;i<len;i++) ans[i]=0;
}
void getksm(int n,ll a[],ll k,ll ans[])
{
    memset(a1,0,sizeof(a1));
    getln(n,a,a2);
    for(int i=0;i<n;i++) a2[i]=a2[i]*k%mod;
    getexp(n,a2,ans);
}
ll read()
{
    ll sum=0,c=getchar();
    while(c<48||c>57) c=getchar();
    while(c>=48&&c<=57) sum=(sum*10+c-48)%mod,c=getchar();
    return sum;
}
int main()
{
	scanf("%d",&n);
    k=read();
	for(int i=0;i<n;i++) scanf("%lld",&a[i]),a[i]%=mod;
	getksm(n,a,k,ans);
	for(int i=0;i<n;i++) printf("%lld ",ans[i]);
}

你可能感兴趣的:(算法)