首页 > 其他分享 >多项式全(?)家桶

多项式全(?)家桶

时间:2022-09-07 18:55:42浏览次数:75  
标签:qpow int 多项式 ll maxn 家桶 ans mod

贴个板子,以备复习

点击查看代码
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<unordered_map>
#include<cmath>
#define mod 998244353
#define maxn 400010
#define ll long long
#define it unordered_map<ll,int>::iterator
using namespace std;
int pos[maxn];
ll A[maxn],B[maxn],C[maxn],w[maxn],D[maxn],E[maxn],P[maxn],Q[maxn],iB[maxn];
ll qpow(ll x,int p){
    ll base,ans;
    for(base=x,ans=1;p;p>>=1,base=base*base%mod){
        if(p&1) ans=ans*base%mod;
    }
    return ans;
}
ll inv(ll x){
    return qpow(x,mod-2);
}
namespace discrete_sqrt{
    unordered_map<long long,int> M;
    int BSGS(ll x){
        int i,N;
        N=(int)(ceil(sqrt(mod-0.5)));
        ll w,w0=qpow(3,N);
        for(i=0,w=1;i<N;++i,w=w*3%mod){
            M[x*w%mod]=i;
        }
        for(i=1,w=w0;i<=N;w=w*w0%mod,++i){
            it t=M.find(w);
            if(t!=M.end()) return i*N-(t->second);
        }
        return -1;
    }
    ll sqr(ll x){
        int u=BSGS(x);
        ll ans=qpow(3,u/2);
        return min(ans,mod-ans);
    }
}
void NTT(ll *L,int N,int type,int lg){
    int i,j,t,d;
    for(i=0;i<N;++i) pos[i]=pos[i>>1]>>1|((1&i)<<lg-1);
    w[0]=1,w[1]=type>0?qpow(3,(mod-1)>>lg):inv(qpow(3,(mod-1)>>lg));
    for(i=2;i<N;++i) w[i]=w[i-1]*w[1]%mod;
    for(i=0;i<N;++i){
        if(pos[i]>i) swap(L[i],L[pos[i]]);
    }
    for(t=1,d=N>>1;t<N;d>>=1,t<<=1){
        for(i=0;i<N;i+=t<<1){
            for(j=0;j<t;++j){
                ll tmp=w[j*d]*L[i+j+t]%mod;
                L[i+j+t]=(L[i+j]+mod-tmp)%mod;
                L[i+j]=(L[i+j]+tmp)%mod;
            }
        }
    }
}
void poly_diff(ll *L,int N){
    for(int i=0;i<N;++i) L[i]=L[i+1]*(i+1)%mod;
    L[N-1]=0;
}
void poly_int(ll *L,int N){
    for(int i=N;i>0;--i) L[i]=L[i-1]*inv(i)%mod;
    L[0]=0;
}
void poly_inv(ll *L1,ll *L2,int N,int lg){
    int i,len,j;
    L2[0]=1;
    for(i=0,len=1;i<=lg;++i,len<<=1){
        for(j=0;j<(len<<1);++j) C[j]=j<len?L1[j]:0;
        NTT(C,len<<1,1,i+1);NTT(L2,len<<1,1,i+1);
        for(j=0;j<(len<<1);++j) L2[j]=(mod+2-L2[j]*C[j]%mod)%mod*L2[j]%mod;
        NTT(L2,len<<1,-1,i+1);
        for(j=0;j<len;++j) L2[j]=L2[j]*inv(len<<1)%mod;
        for(j=len;j<(len<<1);++j) L2[j]=0;
    }
    for(i=N;i<(1<<lg);++i) L2[i]=0;
}
void poly_sqrt(ll *L1,ll *L2,int N,int lg){
    int d,t,i;
    L2[0]=discrete_sqrt::sqr(L1[0]);
    for(t=0,d=1;d<=N;++t,d<<=1){
        for(i=0;i<(d<<1);++i) Q[i]=i<d?L1[i]:0;
        for(i=0;i<(d<<1);++i) iB[i]=0;
        poly_inv(L2,iB,d,t);
        NTT(Q,d<<1,t+1,1);NTT(L2,d<<1,t+1,1);NTT(iB,d<<1,t+1,1);
        for(i=0;i<(d<<1);++i) L2[i]=(C[i]*iB[i]%mod+L2[i])%mod*inv(2)%mod;
        NTT(L2,d<<1,t+1,-1);
        for(i=0;i<(d<<1);++i) L2[i]=L2[i]*inv(d<<1)%mod;
        for(i=d;i<(d<<1);++i) L2[i]=0;
    }
}
void poly_ln(ll *L1,ll *L2,int N,int lg){
    int i;
    for(i=0;i<(N<<1);++i) D[i]=i<N?L1[i]:0;
    poly_diff(D,N);
    for(i=0;i<(N<<1);++i) L2[i]=0;
    poly_inv(L1,L2,N,lg);
    NTT(D,N<<1,1,lg+1);NTT(L2,N<<1,1,lg+1);
    for(i=0;i<(N<<1);++i) L2[i]=L2[i]*D[i]%mod;
    NTT(L2,N<<1,-1,lg+1);
    for(i=0;i<(N<<1);++i) L2[i]=L2[i]*inv(N<<1)%mod;
    poly_int(L2,N);
}
void poly_exp(ll *L1,ll *L2,int N,int lg){
    int i,j,len;
    L2[0]=1;
    for(i=0,len=1;i<=lg;++i,len<<=1){
        poly_ln(L2,E,len,i);
        for(j=len;j<(len<<1);++j) E[j]=0;
        for(j=0;j<(len<<1);++j) C[j]=j<len?L1[j]:0;
        NTT(L2,len<<1,1,i+1);NTT(E,len<<1,1,i+1);NTT(C,len<<1,1,i+1);
        for(j=0;j<(len<<1);++j) L2[j]=(1-E[j]+C[j]+mod)%mod*L2[j]%mod;
        NTT(L2,len<<1,-1,i+1);
        for(j=0;j<(len<<1);++j) L2[j]=L2[j]*inv(len<<1)%mod;
    }
}
void poly_quickpow(ll *L1,ll *L2,int N,int lg,ll k){
    int i;
    poly_ln(L1,P,N,lg);
    for(i=0;i<N;++i) P[i]=P[i]*k%mod;
    poly_exp(P,L2,N,lg);
}
void read(ll &x){
    x=0;
    char c=getchar();
    while(c>'9'||c<'0') c=getchar();
    while(c>='0'&&c<='9'){
        x=(x*10+c-'0')%mod;
        c=getchar();
    }
}
int main(){
    int i,n,m,u;
    ll k;
    scanf("%d",&n);
    read(k);
    for(m=1,u=0;m<n;m<<=1,++u);
    for(i=0;i<n;++i) scanf("%lld",&A[i]);
    poly_quickpow(A,B,m,u,k);
    for(i=0;i<n;++i) printf("%lld ",B[i]);
    system("pause");
    return 0;
}

标签:qpow,int,多项式,ll,maxn,家桶,ans,mod
From: https://www.cnblogs.com/landmine-sweeper/p/16666877.html

相关文章

  • 多项式全家桶
    前置芝士:FTT&NTT不低于高中的数学推导能力不低于高中的代数芝士高等数学初步复变函数初步(?)多项式乘法目标:给定两个多项式\(F,G\),求\(F\timesG\)。有\(F(x)......
  • 生物数据库开发工具GMOD全家桶
    GMOD(GenericModelOrganismDatabase)是专为生物学家创建的开源项目,生物学家用作存储库和工具的交互应用程序和数据库的集合。连通性是GMOD的关键。生物信息学应用程序和......
  • 6-2 多项式求值——15分
    本题要求实现一个函数,计算阶数为n,系数为a[0]…a[n]的多项式(上图)在x点的值。函数接口定义:doublef(intn,doublea[],doublex);其中n是多项式的阶数,a[]中存......
  • 【瞎口胡】多项式牛顿迭代
    前言如果完全不会求导和积分,以及泰勒展开,这里有一个实用性很强的教程3blue1brown-微积分的本质。多项式牛顿迭代给定函数\(G(x)\),求多项式\(F(x)\)使得\(G(F(x))......
  • 【瞎口胡】多项式操作
    前置快速傅里叶变换FFT多项式的基石操作。快速沃尔什变换FWT位运算卷积。鸽了。快速数论变换NTT把FFT搬到了模意义下,终于可以做计数问题啦。多项式牛顿......
  • 多项式
    一、定义在数学中,若干个单项式相加组成的代数式,称为多项式。这里,相加,包括了相减,因为相减,实际是相加他的相反数。单项式,称为多项式的项。单项式的最高次数,称为多项式的......
  • 时隔多年,再次复习 CRT(数论全家桶)
    1.CRT中国剩余定理,用来求解同余方程组\begin{cases}x\equiva_1\pmod{m_1}\\x\equiva_2\pmod{m_2}\\x\equiva_3\pmod{m_3}\\…………\\x\equiva_n\pmod{m......
  • 最小二乘法用于多项式的拟合及程序实现
    改写自:https://blog.csdn.net/piaoxuezhong/article/details/54973750 1#include<stdio.h>2#include"stdlib.h"3#include"math.h"4//#include<vect......
  • 【视频】什么是非线性模型与R语言多项式回归、局部平滑样条、 广义相加GAM分析工资数
    全文链接:http://tecdat.cn/?p=9706原文出处:拓端数据部落公众号在这文中,我将介绍非线性回归的基础知识。非线性回归是一种对因变量和一组自变量之间的非线性关系进行建模......
  • 同余系全家桶
    一.CRT先咕着。二.Lucas定理Lucas定理是用来求这个东西的:\[\dbinom{n}{m}\bmodp\]其中\(p\)为较小质数。其结论为:\[\dbinom{n}{m}\bmodp=\dbinom{\left\l......