多项式全家桶

最新推荐文章于 2023-01-18 16:33:47 发布
最新推荐文章于 2023-01-18 16:33:47 发布
阅读量124 收藏
点赞数
分类专栏： 算法 数论 文章标签： 算法 数学 多项式
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/andyc_03/article/details/117256117
版权
算法 同时被 2 个专栏收录
218 篇文章 1 订阅
订阅专栏
数论
10 篇文章 0 订阅
订阅专栏

1.多项式求逆

【代码】

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define fi first
#define se second
#define lson now<<1
#define rson now<<1|1
typedef long long ll;
const int mod=998244353;
const int maxn=8e5+5;
int a[maxn],b[maxn],c[maxn],n;
int qpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1) res=1LL*res*a%mod;
        a=1LL*a*a%mod;
        b>>=1;
    }
    return res;
}
int rev[maxn];
void ntt(int *A,int lim,int op)
{
    for(int i=0;i<lim;i++)
        if(rev[i]>i) swap(A[i],A[rev[i]]);
    for(int i=1;i<lim;i<<=1)
    {
        int Wn=qpow(3,(mod-1)/(i<<1));
        if(op==-1) Wn=qpow(Wn,mod-2);
        for(int P=i<<1,j=0;j<lim;j+=P)
        {
            int w=1;
            for(int k=0;k<i;k++,w=1LL*w*Wn%mod)
            {
                int x=A[j+k],y=1LL*A[j+k+i]*w%mod;
                A[j+k]=(x+y)%mod;
                A[j+k+i]=(x-y+mod)%mod;
            }
        }
    }
    if(op==-1)
    {
        int inv=qpow(lim,mod-2);
        for(int i=0;i<lim;i++)
            A[i]=1LL*A[i]*inv%mod;
    }
}
void work(int dep,int *a,int *b)
{
    if(dep==1)
    {
        b[0]=qpow(a[0],mod-2);
        return;
    }
    work((dep+1)>>1,a,b);
    int lim=1; int l=0;
    while(lim<(dep+dep)) lim<<=1,l++;
    for(int i=1;i<lim;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=0;i<dep;i++) c[i]=a[i];
    for(int i=dep;i<lim;i++) c[i]=0;
    ntt(c,lim,1); ntt(b,lim,1);
    for(int i=0;i<lim;i++)
        b[i]=1LL*(2-1LL*b[i]*c[i]%mod+mod)%mod*b[i]%mod;
    ntt(b,lim,-1);
    for(int i=dep;i<lim;i++) b[i]=0;
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    work(n,a,b);
    for(int i=0;i<n;i++) printf("%d ",b[i]);
    return 0;
}

 

2.多项式开方

【代码】

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define fi first
#define se second
#define lson now<<1
#define rson now<<1|1
typedef long long ll;
const int maxn=4e5+5;
const int mod=998244353;
int n,f[maxn],ans[maxn],A[maxn],B[maxn],C[maxn],D[maxn];
int rev[maxn],inv2;
int qpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1) res=1LL*res*a%mod;
        a=1LL*a*a%mod;
        b>>=1;
    }
    return res;
}
void ntt(int *A,int lim,int op)
{
    for(int i=0;i<lim;i++)
        if(i<rev[i]) swap(A[i],A[rev[i]]);
    for(int i=1;i<lim;i<<=1)
    {
        int Wn=qpow(3,(mod-1)/(i<<1));
        if(op==-1) Wn=qpow(Wn,mod-2);
        for(int P=i<<1,j=0;j<lim;j+=P)
        {
            int w=1;
            for(int k=0;k<i;k++,w=1LL*w*Wn%mod)
            {
                int x=A[j+k],y=1LL*A[j+k+i]*w%mod;
                A[j+k]=(x+y)%mod;
                A[j+k+i]=(x-y+mod)%mod;
            }
        }
    }
    if(op==-1)
    {
        int inv=qpow(lim,mod-2);
        for(int i=0;i<lim;i++)
            A[i]=1LL*A[i]*inv%mod;
    }
}
void getinv(int *a,int *b,int lim)
{
    b[0]=qpow(a[0],mod-2);
    int limit;
    for(int i=1;i<(lim+lim);i<<=1)
    {
        limit=i<<1;
        for(int j=0;j<i;j++)
            A[j]=a[j],B[j]=b[j];
        for(int j=0;j<limit;j++)
            rev[j]=(rev[j>>1]>>1)|((j&1)?i:0);
        ntt(B,limit,1); ntt(A,limit,1);
        for(int j=0;j<limit;j++)
            b[j]=1LL*(2ll-1LL*A[j]*B[j]%mod+mod)%mod*B[j]%mod;
        ntt(b,limit,-1);
        for(int j=i;j<limit;j++) b[j]=0;
    }
    for(int i=0;i<limit;i++) B[i]=A[i]=0;
    for(int i=lim;i<limit;i++) b[i]=0;
}
void getsqrt(int *a,int *b,int lim)
{
    b[0]=1;
    int limit;
    int *A=C,*B=D;
    for(int i=1;i<(lim+lim);i<<=1)
    {
        limit=i<<1;
        for(int j=0;j<i;j++) A[j]=a[j]; 
        getinv(b,B,i);
        for(int j=1;j<limit;j++)
            rev[j]=(rev[j>>1]>>1)|((j&1)?i:0);
        ntt(A,limit,1); ntt(B,limit,1);
        for(int j=0;j<limit;j++) A[j]=1LL*A[j]*B[j]%mod;
        ntt(A,limit,-1);
        for(int j=0;j<lim;j++) b[j]=1LL*(b[j]+A[j])%mod*inv2%mod;
        for(int j=lim;j<limit;j++) b[j]=0;
    }
    for(int i=0;i<limit;i++) A[i]=B[i]=0;
    for(int i=lim;i<limit;i++) b[i]=0;
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    scanf("%d",&n); inv2=qpow(2,mod-2);
    for(int i=0;i<n;i++) scanf("%d",&f[i]);
    getsqrt(f,ans,n);
    for(int i=0;i<n;i++) printf("%d ",ans[i]);
    return 0;
}

 

3.多项式ln,exp

【代码】

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define fi first
#define se second
#define lson now<<1
#define rson now<<1|1
typedef long long ll;
const int mod=998244353;
const int maxn=8e5+5;
int a[maxn],b[maxn],c[maxn],n;
int ga[maxn],gb[maxn],ans[maxn],f[maxn];
int qpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1) res=1LL*res*a%mod;
        a=1LL*a*a%mod;
        b>>=1;
    }
    return res;
}
int rev[maxn];
void ntt(int *A,int lim,int op)
{
    for(int i=0;i<lim;i++)
        if(rev[i]>i) swap(A[i],A[rev[i]]);
    for(int i=1;i<lim;i<<=1)
    {
        int Wn=qpow(3,(mod-1)/(i<<1));
        if(op==-1) Wn=qpow(Wn,mod-2);
        for(int P=i<<1,j=0;j<lim;j+=P)
        {
            int w=1;
            for(int k=0;k<i;k++,w=1LL*w*Wn%mod)
            {
                int x=A[j+k],y=1LL*A[j+k+i]*w%mod;
                A[j+k]=(x+y)%mod;
                A[j+k+i]=(x-y+mod)%mod;
            }
        }
    }
    if(op==-1)
    {
        int inv=qpow(lim,mod-2);
        for(int i=0;i<lim;i++)
            A[i]=1LL*A[i]*inv%mod;
    }
}
void getinv(int dep,int *a,int *b)
{
    if(dep==1)
    {
        b[0]=qpow(a[0],mod-2);
        return;
    }
    getinv((dep+1)>>1,a,b);
    int lim=1; int l=0;
    while(lim<(dep+dep)) lim<<=1,l++;
    for(int i=1;i<lim;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=0;i<dep;i++) c[i]=a[i];
    for(int i=dep;i<lim;i++) c[i]=0;
    ntt(c,lim,1); ntt(b,lim,1);
    for(int i=0;i<lim;i++)
        b[i]=1LL*(2-1LL*b[i]*c[i]%mod+mod)%mod*b[i]%mod;
    ntt(b,lim,-1);
    for(int i=dep;i<lim;i++) b[i]=0;
}
void getdao(int *a,int *b,int lim)
{
    for(int i=0;i<lim;i++)
        b[i-1]=1LL*i*a[i]%mod;
    b[lim-1]=0;
}
void jifen(int *a,int *b,int lim)
{
    for(int i=1;i<lim;i++)
        b[i]=1LL*a[i-1]*qpow(i,mod-2)%mod;
    b[0]=0;
}
void work(int *a,int *b,int lim)
{
    getdao(a,ga,lim); getinv(lim,a,gb);
    int limit=1; int l=0;
    while(limit<(lim+lim)) limit<<=1,l++;
    for(int i=1;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    ntt(ga,limit,1); ntt(gb,limit,1);
    for(int i=0;i<limit;i++)
        a[i]=1LL*ga[i]*gb[i]%mod;
    ntt(a,limit,-1);
    jifen(a,b,lim);
}
int main()
{
  
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&f[i]);
    int len=1;
    while(len<n) len<<=1;
    work(f,ans,len);
    for(int i=0;i<n;i++) printf("%d ",ans[i]);
    return 0;
}

 

事实证明牛顿迭代虽然看起来比分治NTT少了了个log，但是由于常数较大？导致我的牛顿迭代比分治还慢

【代码】

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define fi first
#define se second
#define lson now<<1
#define rson now<<1|1
typedef long long ll;
const int mod=998244353;
const int maxn=8e5+5;
int c[maxn],n,lnb[maxn];
int ga[maxn],gb[maxn],ans[maxn],f[maxn];
int qpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1) res=1LL*res*a%mod;
        a=1LL*a*a%mod;
        b>>=1;
    }
    return res;
}
int rev[maxn];
void ntt(int *A,int lim,int op)
{
    for(int i=0;i<lim;i++)
        if(rev[i]>i) swap(A[i],A[rev[i]]);
    for(int i=1;i<lim;i<<=1)
    {
        int Wn=qpow(3,(mod-1)/(i<<1));
        if(op==-1) Wn=qpow(Wn,mod-2);
        for(int P=i<<1,j=0;j<lim;j+=P)
        {
            int w=1;
            for(int k=0;k<i;k++,w=1LL*w*Wn%mod)
            {
                int x=A[j+k],y=1LL*A[j+k+i]*w%mod;
                A[j+k]=(x+y)%mod;
                A[j+k+i]=(x-y+mod)%mod;
            }
        }
    }
    if(op==-1)
    {
        int inv=qpow(lim,mod-2);
        for(int i=0;i<lim;i++)
            A[i]=1LL*A[i]*inv%mod;
    }
}
void getinv(int dep,int *a,int *b)
{
    if(dep==1)
    {
        b[0]=qpow(a[0],mod-2);
        return;
    }
    getinv((dep+1)>>1,a,b);
    int lim=1; int l=0;
    while(lim<(dep+dep)) lim<<=1,l++;
    for(int i=1;i<lim;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=0;i<dep;i++) c[i]=a[i];
    for(int i=dep;i<lim;i++) c[i]=0;
    ntt(c,lim,1); ntt(b,lim,1);
    for(int i=0;i<lim;i++)
        b[i]=1LL*(2-1LL*b[i]*c[i]%mod+mod)%mod*b[i]%mod;
    ntt(b,lim,-1);
    for(int i=dep;i<lim;i++) b[i]=0;
}
void getdao(int *a,int *b,int lim)
{
    for(int i=1;i<lim;i++)
        b[i-1]=1LL*i*a[i]%mod;
    b[lim-1]=0;
}
void jifen(int *a,int *b,int lim)
{
    for(int i=1;i<lim;i++)
        b[i]=1LL*a[i-1]*qpow(i,mod-2)%mod;
    b[0]=0;
}
int ta[maxn];
void getln(int *a,int *b,int lim)
{
    for(int i=0;i<(lim<<2);i++) gb[i]=0;
    getdao(a,ga,lim); getinv(lim,a,gb);
    int limit=1; int l=0;
    while(limit<(lim+lim)) limit<<=1,l++;
    for(int i=1;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=lim-1;i<limit;i++) ga[i]=0;
    ntt(ga,limit,1); ntt(gb,limit,1);
    for(int i=0;i<limit;i++)
        ta[i]=1LL*ga[i]*gb[i]%mod;
    ntt(ta,limit,-1);
    jifen(ta,b,lim);
    for(int i=lim;i<limit;i++) b[i]=0;
	b[0]=0;
}
void getexp(int *a,int *b,int lim)
{
    if(lim==1)
    {
        b[0]=1;
        return;
    }
    getexp(a,b,(lim+1)>>1);
    getln(b,lnb,lim);
    int limit=1,l=0;
    while(limit<(lim+lim)) limit<<=1,l++;
    for(int i=1;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=0;i<lim;i++) lnb[i]=(a[i]-lnb[i]+mod)%mod;
    for(int i=lim;i<limit;i++) lnb[i]=b[i]=0;
    lnb[0]++;
    ntt(lnb,limit,1); ntt(b,limit,1);
    for(int i=0;i<limit;i++)
        b[i]=1LL*b[i]*lnb[i]%mod;
    ntt(b,limit,-1);
    for(int i=lim;i<limit;i++) b[i]=0;
}
int main()
{

    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&f[i]);
    getexp(f,ans,n);
    for(int i=0;i<n;i++) printf("%d ",ans[i]);
    return 0;
}

 

 

andyc_03
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
多项式拟合c++程序
01-18
对数据进行多项式拟合，可得拟合多项式的系数，法方程的系数矩阵，平方误差。
多项式全家桶（烂尾不修了，不过板子常数是小的）
最新发布
m0_61179348的博客
10-23 114
一些约定：下面 fi(x)f^i(x)fi(x) 表示 iii 阶导数。f(x)if(x)^if(x)i 表示幂次。若不说明绝大部分除一个多项式时都是代表乘上它的逆。 多项式加减，求导积分 过于简单不讲。(xa)′=axa−1,∫xadx=xa+1a+1+C(x^a)'=ax^{a-1},\displaystyle\int x^a {\rm d}x=\dfrac{x^{a+1}}{a+1}+C(xa)′=axa−1,∫xadx=a+1xa+1​+C。实际操作时 C=0C=0C=0，正确性是好证的。（如果你实
【学习笔记】多项式全家桶
cqbzlydd的博客
01-18 443
终于明白为什么叫多项式全家桶了
[学习笔记] 多项式全家桶 学习笔记
STJqwq的博客
08-20 2610
写在前面：建议初一就学懂这个东西，这是多项式题目的基础。 0.复数类的实现 定义一个复数为 a+bia+bia+bi 。 加法： a+bi+c+di=(a+c)+(b+d)ia+bi+c+di=(a+c)+(b+d)ia+bi+c+di=(a+c)+(b+d)i 减法：同上，系数改成 −1-1−1 乘法：(a+bi)(c+di)=ac+adi+bci+bdi2=(ac−bd)+(ad+bc)i(a+bi)(c+di)=ac+adi+bci+bdi^2=(ac-bd)+(ad+bc)i(a+bi)(c+di)
多项式拟合平滑.docx
06-27
多项式拟合平滑是一种数据分析技术，用于揭示数据集中的潜在趋势和模式。该方法通过构建一个多项式函数，使这个函数尽可能地贴近原始数据点，从而实现对数据的平滑表示。这种方法在处理含有噪声或者不规则变化的数据...
多项式拟合_多项式拟合_VBa_
10-01
在Excel VBA（Visual Basic for Applications）环境中，我们可以利用VBA强大的编程能力来实现复杂的数学运算，例如多项式拟合。多项式拟合是一种通过拟合数据点来创建一个多项式函数的过程，常用于数据分析、预测和...
多项式拟合
01-15
多项式拟合是一种广泛应用在数据分析和预测中的统计方法，它通过构建一个数学模型来近似地表示一组数据点之间的关系。在这个程序中，用户可以选择2到n次的多项式进行拟合，这意味着它可以处理从线性（一次）到高次...
多项式导论
06-29
Introduction to Polynomials　多项式导论 课件
多项式知识小全+全家桶学习小结
a_forever_dream的博客
06-03 971
废话 大概用了两三个星期来搞多项式和生成函数，人都要被虐傻了qwq…… 下面是学到的所有知识的小结，应该是很适合初学者食用的教程，不过一定要按顺序看。 多项式全家桶 快速傅利叶变换FFT 快速数论变换NTT 多项式求逆 学接下来的东西需要的前置知识：微积分的一点初步知识，建议在这里学，看前 444 集大概就够了？泰勒展开与牛顿迭代，以及关于 eee 的一些知识。 多项式ln 多项式exp 多项式开根 多项式除法 多项式多点求值 任意模数NTT 在学的过程中同时也在膜拜师兄的优秀板子，所以每一篇的码风不尽相同
多项式全家桶 基础知识+各种操作+例题（细节证明）
DD(XYX)的硫氰化氘博客
02-19 6204
只讲了一些基本操作，过于深奥的多项式三角函数、快速插值可以出门看 OneInDark 的博客
进军多项式（一）：多项式全家桶模板
jwg2732的专栏
02-09 4093
多项式乘法(NTT) 默认大家会了FFT…… 我们知道FFT中因为要用三角函数和复数，所以对精度要求高 并且常数比较大 那么有没有可以代替单位根的呢？ 我们思考一下在FFT中应用的单位根的性质 1:wn0=1w_n^0=1wn0​=1 保证了有初始值 2:wni+k=wniwnjw_n^{i+k}=w_n^iw_n^jwni+k​=wni​wnj​,也就是wnk=(wn1)kw_n^k=(w_n^1)^kwnk​=(wn1​)k 因此只要求出一个就可以递推 3:wnk+n/2=−wnkw_n^{k+n/2}
NTT笔记和多项式全家桶
qq_54026781的博客
11-21 843
NTT和多项式全家桶模板
【模板】多项式全家桶
weixin_33912638的博客
03-21 141
FFT struct Z { double x,y; Z(double _x=0,double _y=0):x(_x),y(_y){}; }; Z operator +(Z x,Z y) {return Z(x.x+y.x,x.y+y.y);} Z operator -(Z x,Z y) {return Z(x.x-y.x,x.y-y.y);} Z operator *(Z x,Z...
【学习笔记】多项式全家桶（包含全套证明）
热门推荐
繁凡さん的博客
12-29 1万+
目录0x00.多项式基本概述0x01.多项式的基础概念0x02. 多项式的度0x03. 多项式的乘法0x04. 多项式的逆元0x10.拉格朗日插值0x20.多项式乘法0x21.快速傅里叶变换（FFT）0x22.快速数论变换（NTT）0x23.快速沃尔什变换（FWT）0x24.快速莫比乌斯变换（FMT）0x30.多项式求逆0x40.多项式开方0x50.多项式除法 / 取模0x60.多项式牛顿迭代0x70.多项式多点求值 / 快速插值0x80.多项式三角函数0x90.多项式反...
C语言课程设计一元多项式计算系统
首先提示用户输入两个多项式A和B，输入的形式是先输入多项式A的项数，然后回车，接着输入多项式A的第一项的系数，每一项按照先输入系数，然后输入空格或回车，再输入多项式A的第一项的指数，然后继续输入多项式A的...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值