多项式模板

最新推荐文章于 2021-02-04 10:35:40 发布
最新推荐文章于 2021-02-04 10:35:40 发布
阅读量224 收藏 1
点赞数
分类专栏： 数论
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_35950004/article/details/85370765
版权

检查数据:
input:
6
1 2 3 4 5 6
1.NTT多项式求逆

#include<bits/stdc++.h>
#define maxn 2100005
#define LL long long
#define mod 998244353
using namespace std;

int lg[maxn],r[maxn],w[maxn]={1},wlen;
int Pow(int base,int k)
{	int ret=1;
	for(;k;k>>=1,base=1ll*base*base%mod) if(k&1) ret=1ll*ret*base%mod;
	return ret;}
inline void NTT(int A[maxn],int n,int tp)//n=2^k
{
	int lgn = lg[n];
	for(int i=1;i<n;i++) r[i] = (r[i>>1]>>1)|((i&1)<<(lgn-1));
	for(int i=1;i<n;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int L=2;L<=n;L<<=1)
		for(int st=0,l=L>>1,inc=wlen/l;st<n;st+=L)
			for(int k=st,x=0;k<st+l;k++,x+=inc)
			{	int tmp = 1ll *(tp==1?w[x]:w[2*wlen-x])*A[k+l] % mod;
				A[k+l]=(A[k]-tmp)%mod,A[k]=(A[k]+tmp)%mod;}
	if(tp==-1) for(int i=0,inv=Pow(n,mod-2);i<n;i++) A[i]=1ll*A[i]*inv%mod;
}
void Inv(int A[maxn],int B[maxn],int n)
{	B[0] = Pow(A[0],mod-2);
	static int tmp[maxn];
	for(int i=2;i<(n<<1);i<<=1)// mod x^i
	{	for(int j=0;j<i;j++) tmp[j] = j<n?A[j]:0;
		NTT(tmp,i<<1,1),NTT(B,i<<1,1);
		for(int j=0;j<(i<<1);j++) B[j]=1ll*B[j]*(2-1ll*B[j]*tmp[j]%mod)%mod;
		NTT(B,i<<1,-1);	for(int j=min(n,i);j<(i<<1);j++) B[j] = 0;}
}

int a[maxn],b[maxn];
int main()
{
	int n;
	scanf("%d",&n);
	for(wlen=1;(n<<1)>(wlen<<1);wlen<<=1);
	for(int i=1,pw=Pow(3,(mod-1)/(2*wlen));i<=2*wlen;i++) w[i] = 1ll * w[i-1] * pw % mod;
	for(int i=2;i<=2*wlen;i++) lg[i] = lg[i>>1]+1;
	for(int i=0;i<n;i++) scanf("%d",&a[i]);
	Inv(a,b,n);
	for(int i=0;i<n;i++) printf("%d%c",(b[i]+mod)%mod,i==n-1?'\n':' ');
}

output:

2.多项式除法：
感觉我好短啊。

#include<bits/stdc++.h>
#define maxn 2100005
#define LL long long
#define mod 998244353
using namespace std;

int lg[maxn],r[maxn],w[maxn]={1},wlen;
int Pow(int base,int k)
{	int ret=1;
	for(;k;k>>=1,base=1ll*base*base%mod) if(k&1) ret=1ll*ret*base%mod;
	return ret;}
inline void NTT(int A[maxn],int n,int tp)//n=2^k
{	int lgn = lg[n];
	for(int i=1;i<n;i++) r[i] = (r[i>>1]>>1)|((i&1)<<(lgn-1));
	for(int i=1;i<n;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int L=2;L<=n;L<<=1)
		for(int st=0,l=L>>1,inc=wlen/l;st<n;st+=L)
			for(int k=st,x=0;k<st+l;k++,x+=inc)
			{	int tmp = 1ll *(tp==1?w[x]:w[2*wlen-x])*A[k+l] % mod;
				A[k+l]=(A[k]-tmp)%mod,A[k]=(A[k]+tmp)%mod;}
	if(tp==-1) for(int i=0,inv=Pow(n,mod-2);i<n;i++) A[i]=1ll*A[i]*inv%mod;}
void mul(int A[maxn],int B[maxn],int C[maxn],int n,int m)
{	int lgn = lg[n+m]+1,len = 1<<lgn;
	static int sta[maxn];
	for(int i=0;i<len;i++) sta[i]=i<=n?A[i]:0,C[i]=i<=m?B[i]:0;
	NTT(sta,len,1),NTT(C,len,1);
	for(int i=0;i<len;i++) C[i] = 1ll * C[i] * sta[i] % mod;
	NTT(C,len,-1);}
void Inv(int A[maxn],int B[maxn],int n)
{	B[0] = Pow(A[0],mod-2);
	static int tmp[maxn];
	for(int i=2;i<(n<<1);i<<=1)// mod x^i
	{	for(int j=0;j<i;j++) tmp[j] = j<n?A[j]:0;
		NTT(tmp,i<<1,1),NTT(B,i<<1,1);
		for(int j=0;j<(i<<1);j++) B[j]=1ll*B[j]*(2-1ll*B[j]*tmp[j]%mod)%mod;
		NTT(B,i<<1,-1);	for(int j=min(n,i);j<(i<<1);j++) B[j] = 0;}}
void Div(int A[maxn],int B[maxn],int C[maxn],int R[maxn],int n,int m)
{	static int ra[maxn],rb[maxn];
	for(int i=0,lim=max(n,m);i<=lim;i++) ra[i]=i<=n?A[n-i]:0,rb[i]=i<=m?B[m-i]:0;
	Inv(rb,C,n-m+1),mul(ra,C,C,n,n-m);
	for(int i=0;i<=2*n-m;i++) if(i<n-m-i)swap(C[i],C[n-m-i]);else if(i>n-m)C[i]=0;
	mul(C,B,R,n-m,m);for(int i=0;i<=n;i++) R[i]=i<m?(A[i]-R[i])%mod:0;}
void Init(int n)
{	for(wlen=1;n>=(wlen<<1);wlen<<=1);
	for(int i=1,pw=Pow(3,(mod-1)/(2*wlen));i<=2*wlen;i++) w[i]=1ll*w[i-1]*pw%mod;
	for(int i=2;i<=2*wlen;i++) lg[i] = lg[i>>1] + 1;}
int a[maxn],b[maxn],c[maxn],R[maxn];
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<=n;i++) scanf("%d",&a[i]);
	for(int i=0;i<=m;i++) scanf("%d",&b[i]);
	Init(n<<1);
	Div(a,b,c,R,n,m);
	for(int i=0;i<=n-m;i++) printf("%d%c",(c[i]+mod)%mod,i==n-m?'\n':' ');
	for(int i=0;i<m;i++) printf("%d%c",(R[i]+mod)%mod,i==m-1?'\n':' ');
}

3.多项式exp
AC Code:（PS ： 一旦到了需要多次调用函数的时候很多细节（清零）问题就出来了，以上的代码好像都没有考虑的说。。。。。。求逆，求 L n Ln Ln，求 e x p exp exp都以本代码为准）
63行还是有点短的说。

#include<bits/stdc++.h>
#define maxn 300005
#define mod 998244353
using namespace std;

int lg[maxn],r[maxn],w[maxn]={1},wlen,inv[maxn]={1,1};
int Pow(int base,int k)
{	int ret = 1;
	for(;k;k>>=1,base=1ll*base*base%mod) if(k&1) ret=1ll*ret*base%mod;
	return ret; }
void Init(int n)
{	for(wlen=1;n>=(wlen*2);wlen<<=1);
	for(int i=1,pw=Pow(3,(mod-1)/(2*wlen));i<=2*wlen;i++) w[i] = 1ll * w[i-1] * pw % mod;
	for(int i=2;i<=2*wlen;i++) lg[i] = lg[i>>1] + 1; 
	for(int i=2;i<=2*wlen;i++) inv[i] = 1ll * (mod - mod / i) * inv[mod % i] % mod; }
inline void NTT(int *A,int n,int tp)
{	int lgn = lg[n];
	for(int i=1;i<n;i++) r[i] = (r[i>>1]>>1)|((i&1)<<(lgn-1));
	for(int i=1;i<n;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int L=2;L<=n;L<<=1)
		for(int st=0,l=L>>1,inc=wlen/l;st<n;st+=L)
			for(int k=st,x=0;k<st+l;k++,x+=inc)
			{ 	int tmp = 1ll * (tp==1?w[x]:w[2*wlen-x]) * A[k+l] % mod;
				A[k+l]=(A[k]-tmp)%mod,A[k]=(A[k]+tmp)%mod;}
	if(tp==-1) for(int i=0,inv=Pow(n,mod-2);i<n;i++) A[i]=1ll*A[i]*inv%mod;}
void Inv(int *A,int *B,int n)
{	B[0] = Pow(A[0] , mod-2);
	static int tmp[maxn];
	for(int k=2;k<(n<<1);k<<=1)
	{	for(int i=0,lim=min(k,n);i<(k<<1);i++) tmp[i]=i<lim?A[i]:0,B[i]=i<lim?B[i]:0;
		NTT(tmp,k<<1,1),NTT(B,k<<1,1);
		for(int i=0,lim=k<<1;i<lim;i++) B[i]=1ll*B[i]*(2-1ll*B[i]*tmp[i]%mod)%mod;
		NTT(B,k<<1,-1);for(int i=min(k,n);i<(k<<1);i++) B[i] = 0;}
}
void cLn(int *A,int *B,int n)
{	Inv(A,B,n);
	static int tmp[maxn];
	int lgn = lg[n-1]+2 , len = 1<<lgn;
	for(int i=0;i<len;i++) tmp[i]=i<n-1?1ll*A[i+1]*(i+1)%mod:0;
	NTT(tmp,len,1),NTT(B,len,1);
	for(int i=0;i<len;i++) tmp[i]=1ll*B[i]*tmp[i]%mod;
	NTT(tmp,len,-1);B[0]=0;
	for(int i=1;i<n;i++) B[i] = 1ll * tmp[i-1] * inv[i] % mod; 
	for(int i=n;i<len;i++) B[i] = 0;}
void eXp(int *A,int *B,int n)
{	B[0] = 1;
	static int tmp[maxn];
	for(int k=2;k<(n<<1);k<<=1)
	{	cLn(B,tmp,k);
		for(int i=0,lim=min(n,k);i<(k<<1);i++) tmp[i]=i<lim?((i==0)-tmp[i]+A[i])%mod:0,B[i]=i<lim?B[i]:0;
		NTT(B,k<<1,1),NTT(tmp,k<<1,1);
		for(int i=0;i<(k<<1);i++) B[i]=1ll*B[i]*tmp[i]%mod;
		NTT(B,k<<1,-1);for(int i=min(n,k);i<(k<<1);i++) B[i]=0;}}
int a[maxn],b[maxn];
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++) scanf("%d",&a[i]);
	Init(n<<1);
	eXp(a,b,n);
	for(int i=0;i<n;i++) printf("%d%c",(b[i]+mod)%mod,i==n-1?'\n':' ');
}

4多项式开根

#include<bits/stdc++.h>
#define maxn 300005
#define mod 998244353
using namespace std;

int w[maxn]={1},lg[maxn],r[maxn],wlen;
int Pow(int base,int k)
{	int ret = 1;
	for(;k;k>>=1,base=1ll*base*base%mod) if(k&1) ret=1ll*ret*base%mod;
	return ret;}
void Init(int n)
{	for(wlen=1;n>=2*wlen;wlen<<=1);
	for(int i=1,pw=Pow(3,(mod-1)/(2*wlen));i<=2*wlen;i++) w[i]=1ll*w[i-1]*pw%mod;
	for(int i=2;i<=2*wlen;i++) lg[i]=lg[i>>1]+1;}
inline void NTT(int *A,int n,int tp)
{	int lgn = lg[n];
	for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lgn-1));
	for(int i=1;i<n;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int L=2;L<=n;L<<=1)
		for(int st=0,l=L>>1,inc=wlen/l;st<n;st+=L)
			for(int k=st,x=0,tmp;k<st+l;k++,x+=inc)
				tmp=1ll*(tp==1?w[x]:w[2*wlen-x])*A[k+l]%mod,
				A[k+l]=(A[k]-tmp)%mod,A[k]=(A[k]+tmp)%mod;
	if(tp==-1) for(int i=0,inv=Pow(n,mod-2);i<n;i++) A[i]=1ll*A[i]*inv%mod;}
void Inv(int *A,int *B,int n)
{	B[0]=Pow(A[0],mod-2),B[1]=B[2]=B[3]=0;
	static int tmp[maxn];
	for(int k=2;k<(n<<1);k<<=1)
	{	for(int i=0;i<(k<<1);i++) tmp[i]=i<k?A[i]:0,B[i]=i<k?B[i]:0;
		NTT(tmp,k<<1,1),NTT(B,k<<1,1);
		for(int i=0;i<(k<<1);i++) B[i]=1ll*B[i]*(2-1ll*B[i]*tmp[i]%mod)%mod;
		NTT(B,k<<1,-1);for(int i=min(k,n);i<(k<<1);i++) B[i]=0;}
}
void Sqt(int *A,int *B,int n)
{	B[0]=1,B[1]=B[2]=B[3]=0;
	static int tmp[2][maxn];
	for(int k=2;k<(n<<1);k<<=1)
	{	for(int i=0;i<(k<<1);i++) tmp[0][i]=i<k?A[i]:0,B[i]=i<k?B[i]:0;
		Inv(B,tmp[1],k),NTT(tmp[0],k<<1,1),NTT(tmp[1],k<<1,1),NTT(B,k<<1,1);
		for(int i=0;i<(k<<1);i++) B[i] = 1ll * (mod+1)/2 * tmp[1][i] % mod * (tmp[0][i]+1ll*B[i]*B[i]%mod) % mod; 
		NTT(B,k<<1,-1);for(int i=k;i<(k<<1);i++) B[i]=0;}}
int n,m;
int a[maxn],b[maxn],c[maxn],d[maxn];
int main()
{
	scanf("%d%d",&n,&m);m++;
	for(int i=0,x;i<n;i++) scanf("%d",&x),a[x]++;
	Init(m<<1);
	for(int i=0;i<=m;i++) b[i] = (i==0) - 4ll * a[i];
	Sqt(b,c,m);	
	c[0]++;
	Inv(c,d,m);
	for(int i=1;i<m;i++) 
		printf("%d\n",(2ll*d[i]%mod+mod)%mod);
}

5.多项式求幂
有3种算法：（后两种只能玩整数幂）
1.多项式exp O ( n log ⁡ n ) ∗ 30 O(n \log n) * 30 O(nlogn)∗30
2.多项式快速幂，每层3次FFT T ( n ) = T ( n / 2 ) + O ( l e n log ⁡ l e n ) ， 实 际 是 O ( l e n log ⁡ l e n log ⁡ n ) T(n) = T(n/2) + O(len \log len)，实际是O(len \log len \log n) T(n)=T(n/2)+O(lenloglen)，实际是O(lenloglenlogn)
3.多项式快速幂，NTT后对每个数单独进行快速幂然后再NTT回来。 O ( n log ⁡ n ) O(n \log n) O(nlogn)
关于这3者的区别可以看这篇博客
Codeforces 1096G Lucky Tickets

Freopen
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
【多项式n连】各种多项式模板（从入门到入土）
shiveringkonnyaku的博客
03-04 1683
qwq。。。好像很久没有写博客了的样子。。。因为最近一段时间一直处在觉得自己太蒻了的焦虑当中。。。导致啥也没学进去。。。这样的。。。本来定下的肝高数和概率论的计划也因为各种各样的原因没能完成。。。而降低到了只是零散地又复习了一些数学。。。然后发现luogu上有很多多项式的板子。。。就去做了一点。。。这样子。。。然后发现好久没更博客了。。。就准备放上来一些。。。也算是水一水博客了。。。反正老年选手已...
稀疏多项式
12-17
稀疏多项式是一种在计算机科学和数学中处理大型多项式表达式时常用的数据结构，尤其在算法和计算复杂性理论中扮演着重要角色。在实际应用中，如果一个多项式的系数大多数为零，那么使用传统的数组表示法会浪费大量...
多项式模板&题目整理
banchangyue3928的博客
11-27 248
注：多项式的题目，数组应开：N的最近2的整数次幂的4倍。 多项式乘法 FFT模板 时间复杂度\(O(n\log n)\)。 模板： void FFT(Z *a,int x,int K){ static int rev[N],lst; int n=(1<<x); if(n!=lst){ for(int i=0;i<n;i++)rev...
多项式总结及模板
caoyang1123的博客
03-30 444
1.求逆 设要对多项式A求逆，逆为B： 求在模ceil（n/2）的逆，将原式与之相减后平方，发现可得在模n意义下也是0了，然后两边同乘A，移项即可倍增了（注意此时A也是在模当前长度意义下的）。 最后倍增形式为B'=2*B-B*B*A （b’等于2b减b方a），边界为常数取逆元，也可发现多项式有无逆取决于常数有无逆元 2.开根 与求逆类似，先求ceil（n/2）,然后将A移到左边使等式右...
c++模板实现多项式相加
03-03
本话题将详细讲解如何利用C++模板实现链表表示的多项式相加。 首先，我们需要了解多项式的基本概念。一个多项式是由常数、变量及其指数组合而成的数学表达式，例如2x^3 + 5x^2 - 3x + 7。在C++中，我们可以使用链表...
大景深系统多项式相位模板的优化设计
01-11
大景深系统多项式相位模板的优化设计，王晋疆，韩瑞雨，基于波前编码大景深成像系统的立方相位模板，提出了一种新的奇对称多项式相位模板。并根据不同离焦MTF和聚焦MTF之间的希尔伯特空间
多项式拟合源代码
08-08
对于C++，我们可以使用标准模板库(STL)中的向量和迭代器，以及数值库如Eigen或BLAS/LAPACK来实现。Python中，可以利用Numpy或Scipy库的`polyfit`函数进行多项式拟合。 在"多项式拟合源代码"的压缩包中，`...
利用单链表实现多项式基本运算(C++模板实现）
08-10
本主题聚焦于如何使用C++模板实现单链表，并基于此实现多项式的加、减、乘等基本运算。下面将详细介绍这两个知识点。 **1. C++模板** C++模板是一种泛型编程工具，它允许我们编写可以处理不同数据类型的代码。模板...
多项式系列模板
Never give in.
04-23 711
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 ...
多项式模板QAQ
cgh_Andy's Blog
03-06 407
失踪人口回归？ 似乎抄了挺多天的多项式题目代码。。留几个模板吧。。 （都不知道是不是好模板。。 反正也是自己看 最长不超过80行我去 UOJ #34. 多项式乘法 #include &lt;bits/stdc++.h&gt; #define me(a,x) memset(a,x,sizeof a) using namespace std; const int N=3e5+2,inf=1...
多项式模板（inv, sqrt, ln, exp）
lifehappy
02-04 1364
#include <bits/stdc++.h> using namespace std; const int mod = 998244353, inv2 = mod + 1 >> 1; namespace Quadratic_residue { struct Complex { int r, i; Complex(int _r = 0, int _i = 0) : r(_r), i(_i) {} }; int I2; Complex op
多项式求解【模板】
温文尔雅
11-23 315
#include<iostream> #include<cstdio> #include<cstring> #include<string.h> #include<algorithm> #include<vector> #include<cmath> #include<stdlib.h> #include<time.h> #include<stack> #include<set> #include<
NTT和多项式各种基础运算模板
wzf_2000的博客
06-03 660
#include #define mod 998244353 #define N 400009 #define ll long long using namespace std; ll w[2][N],inv[N]; ll A[N],B[N],C[N],D[N],E[N],F[N]; ll W,G; ll pow(ll a,ll b,ll c) { ll ans=1; for (;b;a=a
多项式算法模板
romiqi
07-26 111
讲解link 附全套模板： div除法，取模 #include<bits/stdc++.h> #define poly vector<int> #define ll long long #define mod 998244353 using namespace std; inline int read(){ int res=0,f=1;char ch=getchar();...
[多项式ln][多项式exp][多项式求幂][生成函数][DP][FNT] BZOJ 3684: 大朋友和多叉树
Vectorxj的博客
03-06 790
SolutionSolutionSolution 把DP写成生成函数的形式。f(x)=x+∑d∈Dfd(x)f(x)=x+∑d∈Dfd(x)f(x)=x+\sum_{d\in D}f^d(x)设g(f(x))=xg(f(x))=xg\big(f(x)\big)=x，有g(f(x))g(x)==f(x)−∑d∈Dfd(x)x−∑d∈Dxdg(f(x))=f(x)−∑d∈Dfd(x)g(x)=x−∑...
将不同的边缘检测算子的实验结果进行比较，如下图所示：从图中可以看出，Sobel算子和Prewitt算子能够检测出较为清晰的边缘线，而Roberts算子和Laplacian算子的效果相对较差。这是因为Sobel算子和Prewitt算子采用的是多项式模板，能够更加准确地计算图像梯度，而Roberts算子和Laplacian算子采用的是线性模板，对于图像中的噪声和平滑区域容易产生误检。
最新发布
05-30
相比之下，Sobel算子和Prewitt算子的效果确实要比Roberts算子和Laplacian算子更好，这也验证了多项式模板的优越性。另外，对于Laplacian算子的效果相对较差，我们可以使用高斯滤波器来平滑图像，以提高Laplacian算子...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值