多项式板子 [vector＜int＞版本]

于 2023-02-20 11:05:27 发布
阅读量201 收藏
点赞数
分类专栏： 板子 文章标签： c++ 算法 c语言
原文链接：https://codeforces.com/submissions/MoonLight__
版权

偷的板子
跑起来非常非常快。

#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>

using namespace std;

int read(){
	int res = 0 , flag = 1 ;
	char c = getchar() ;
	while(!isdigit(c)){
		if(c == '-') flag = -1 ;
		c = getchar() ;
	}
	while(isdigit(c)){
		res = (res << 1) + (res << 3) + (c ^ 48) ;
		c = getchar() ;
	}
	return res * flag ; 
}

void write(int x){
	if(x < 0) {
		putchar('-') ;
		x = - x ;
	}
	if(x >= 10) write(x / 10) ;
	putchar('0' + x % 10) ;
}

void write(int x , char c){
	write(x) ;
	putchar(c) ; 
}

// #define debug(x) cout<<"[debug]"#x<<"="<<x<<endl
typedef long long ll;
typedef long double ld;
typedef pair<int,int> pii;
const double eps=1e-8;
const int inf=0x3f3f3f3f;

#ifndef ONLINE_JUDGE
#define debug(...)
// #include<debug>
#else
#define debug(...)
#endif

const int N= 1000006;
namespace NTT
{
    typedef vector<ll> Poly;
    const int mod=998244353,G=3,Gi=332748118;
    const int M=2e6+1e5;
    int bit,tot;
    int rev[M];
    int inv[M];
    int init_inv=[](){
    inv[0]=inv[1]=1;
    for(int i=2;i<M;i++)
        inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
    return 0;
    }();
    ll qmi(ll a,ll b,ll p)
    {
        ll res=1;
        while(b)
        {
            if(b&1)
            res=res*a%p;

            a=a*a%p;
            b>>=1;
        }
        return res;
    }
    void NTT(Poly &a,int inv)
    {
        for(int i=0;i<tot;i++)
            if(i<rev[i]) swap(a[i],a[rev[i]]);
    
        for(int mid=1;mid<tot;mid*=2)
        {
            ll w1=qmi(inv==1?G:Gi,(mod-1)/(2*mid),mod);
            for(int i=0;i<tot;i+=mid*2)
            {
                ll wk=1;
                for(int j=0;j<mid;j++,wk=wk*w1%mod)
                {
                    ll x=a[i+j];
                    ll y=wk*a[i+j+mid]%mod;
                    a[i+j]=(x+y)%mod;
                    a[i+j+mid]=(x-y+mod)%mod;
                }
            }
        }
    
        if(inv==-1)//就不用后面除了
        {
            ll intot=qmi(tot,mod-2,mod);
            for(int i=0;i<tot;i++)
            {
                a[i]=a[i]*intot%mod;
            }
        }
    }
    Poly mul(Poly a,Poly b)//deg是系数的数量，所以有0~deg-1次项
    {
        int deg=(int)a.size()+b.size()-1;
        bit=0;
        while((1<<bit)<deg) bit++;//至少要系数的数量
        tot=1<<bit; //系数项为0~tot-1
        for(int i=0;i<tot;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));

        Poly c(tot);
        a.resize(tot),b.resize(tot);
        NTT(a,1),NTT(b,1);
        for(int i=0;i<tot;i++) c[i]=a[i]*b[i]%mod;
        NTT(c,-1);
        // c.resize(1000001);
        return c;
    }
    Poly operator *(Poly &a,Poly &b)
    {
        return mul(a,b);
    }
    Poly operator *(Poly &a,int &t)
    {
        Poly res;
        for(int i=0;i<a.size();i++) res.push_back(1ll*a[i]*t%mod);
        return res;
    }
    Poly operator *(Poly &a,ll &t)
    {
        Poly res;
        for(int i=0;i<a.size();i++) res.push_back(a[i]*t%mod);
        return res;
    }
    Poly operator +(Poly &a,Poly &b)
    {
        Poly res(a);
        res.resize(max(a.size(),b.size()));
        for(int i=0;i<b.size();i++) res[i]=(res[i]+b[i])%mod;
        return res;
    }
    Poly operator -(Poly &a,Poly &b)
    {
        Poly res(a);
        res.resize(max(a.size(),b.size()));
        for(int i=0;i<b.size();i++) res[i]=(res[i]-b[i]+mod)%mod;
        return res;
    }
    Poly Inv(Poly &f,int deg)//多项式f对于x^deg的逆元(注意rev[]等数组要开到2*deg的空间级别,f[]要开deg级别)
    {
        if(deg==1) return Poly(1,qmi(f[0],mod-2,mod));

        Poly B=Inv(f,(deg+1)>>1);//上一个逆元
        Poly A(f.begin(),f.begin()+deg);

        bit=0;
        while((1<<bit)<(deg<<1)) bit++;//至少要系数的数量
        tot=1<<bit; //系数项为0~tot-1
        for(int i=0;i<tot;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
        
        A.resize(tot),B.resize(tot);
        NTT(A,1),NTT(B,1);
        for(int i=0;i<tot;i++)
        A[i]=B[i]*(2-A[i]*B[i]%mod+mod)%mod;
        NTT(A,-1);
        A.resize(deg);
        return A;
    }
    Poly cdq_ntt(int l,int r,vector<Poly> &f)
    {
        if(l==r) return f[l];
        int mid=l+r>>1;
        return mul(cdq_ntt(l,mid,f),cdq_ntt(mid+1,r,f));
    }
    namespace Cipolla
    {
        int W;
        struct cp
        {
            ll x,y;
            cp(ll x=0,ll y=0): x(x),y(y) {}
            cp operator+ (const cp _)const{
                return {(x+_.x)%mod,(y+_.y)%mod};
            } 
            cp operator* (const cp _)const{
                return {(x*_.x%mod+y*_.y%mod*W%mod)%mod,(x*_.y+y*_.x)%mod};
            }
        };
        cp qmi(cp a,int b)
        {
            cp res=1;
            while(b)
            {
                if(b&1) res=res*a;

                a=a*a;
                b>>=1;
            }
            return res;
        }
        int cipolla(int n)
        {
            if(qmi(n,(mod-1)>>1).x!=1)
            {
                return -1;
            }
            int a=rand()%mod;
            while(!a||qmi((1ll*a*a-n+mod)%mod,(mod-1)>>1).x==1)
            {
                a=rand()%mod;
            }
            W=(1ll*a*a-n+mod)%mod;

            int x=qmi(cp(a,1),(mod+1)>>1).x;
            if(x>mod-x) x=mod-x;
            return x;
        }
    }
    Poly Sqrt(Poly &A,int deg)
    {
        if(deg==1)
        {
            return {Cipolla::cipolla(A[0])};//A[0]=1修改这里即可
        }

        Poly f0=Sqrt(A,(deg+1)>>1);
        f0.resize(deg);
        Poly inf0=Inv(f0,deg);
        Poly temp(A.begin(),A.begin()+deg);
        temp=mul(inf0,temp);
        temp.resize(deg);
        for(int i=0;i<deg;i++) f0[i]=(f0[i]+temp[i])*inv[2]%mod;

        return f0;
    }
    Poly Deriv(Poly f)//求导
    {
        for(int i=0;i<f.size();i++) f[i]=f[i+1]*(i+1)%mod;
        f.pop_back();
        return f;
    }
    Poly Integ(Poly f)//积分
    {
        f.push_back(0);
        for(int i=f.size()-1;i>=1;i--) f[i]=f[i-1]*inv[i]%mod;
        f[0]=0;
        return f;
    }
    Poly Ln(Poly f,int deg)//f[0]=1
    {
        f=mul(Deriv(f),Inv(f,deg));
        f.resize(deg-1);
        return Integ(f);
    }
    Poly Exp(Poly &A,int deg)//A[0]=0，记得至少开deg大小
    {
        if(deg==1) return {1};
        Poly f0=Exp(A,(deg+1)>>1);
        Poly temp=f0;
        temp.resize(deg);
        temp=Ln(temp,deg);
        for(int i=0;i<deg;i++) temp[i]=(A[i]-temp[i]+mod)%mod;
        temp[0]=(temp[0]+1)%mod;
        temp=mul(f0,temp);
        temp.resize(deg);
        return temp;
    }
    Poly qmi_ntt(Poly A,int k,int deg=-1)//A(x)^k且A[0]=1
    {
        if(deg==-1) deg=(A.size()-1)*k+1;
        A.resize(deg);
        A=Ln(A,deg);
        A=A*k;
        A=Exp(A,deg);
        return A;
    }
    Poly qmi_ntt(Poly A,string s,int deg=-1)//A(x)^k
    {
        ll k=0;//mod (p)
        ll kk=0;//mod (p-1),即phi(p)
        bool ty=false;
        for(int i=0;i<s.size();i++)
        {
            k=(1ll*k*10+(s[i]-'0'));
            if(k>=mod) k%=mod,ty=true;
            kk=(1ll*kk*10+(s[i]-'0'))%(mod-1);
        }
        if(deg==-1) deg=(A.size()-1)*k+1;
        A.resize(deg);
        int n=A.size();
        int t=0;
        while(t<n&&!A[t]) t++;

        if(t==n||ty&&t>=1)
        {
            return Poly(n,0);
        }
        ll temp=qmi(A[t],mod-2,mod);
        Poly B;
        for(int i=t;i<A.size();i++) B.push_back(A[i]*temp%mod);
        B=Ln(B,B.size());
        B=B*k;
        B=Exp(B,B.size());
        B.resize(n);
        Poly res(n);
        temp=qmi(A[t],kk,mod);
        for(ll i=1ll*t*k;i<n;i++) res[i]=B[i-1ll*t*k]*temp%mod;
        return res;
    }
    Poly psqmi_ntt(Poly A,int k)
    {
        Poly res(1,1);
        while(k)
        {
            if(k&1) res=res*A;
            A=A*A;
            k>>=1;
        }
        return res;
    }
}
using namespace NTT;

int main()
{
	/* P1919 A*B
	string a , b ;
	cin >> a >> b ;
	int n = a.size() , m = b.size() ;
	Poly A(n) , B(m) ;
	for(int i = 0 ; i < n ; i ++) A[i] = a[n - i - 1] - '0' ;
	for(int i = 0 ; i < m ; i ++) B[i] = b[m - i - 1] - '0' ;
	Poly c = A * B ;
	for(int i = 0 ; i < c.size() ; i ++){
		if(c[i] >= 10) c[i + 1] += c[i] / 10 , c[i] %= 10 ;
	}
	while(c.size() > 1 && c.back() == 0) c.pop_back() ;
	for(int i = c.size() - 1 ; i >= 0 ; i --)
		write(c[i]) ; 
	*/
	
	
	/* P4238 【模板】多项式乘法逆
		int n = read() ;
		Poly A(n) ;
		for(int i = 0 ; i < n ; i ++) A[i] = read() ;
		Poly B = Inv(A , n) ;
		for(int i = 0 ; i < n ; i ++) write(B[i] , ' ') ; 
	*/
	
	/* P5205 【模板】多项式开根
		int n = read() ;
		Poly A(n) ;
		for(int i = 0 ; i < n ; i ++) A[i] = read() ;
		Poly B = Sqrt(A , n) ; 
		for(int i = 0 ; i < n ; i ++) write(B[i] , ' ') ;
	*/
	
	/* P4725 【模板】多项式对数函数（多项式 ln）
		int n = read() ;
		Poly A(n) ;
		for(int i = 0 ; i < n ; i ++) A[i] = read() ;
		Poly B = Ln(A , n) ; 
		for(int i = 0 ; i < n ; i ++) write(B[i] , ' ') ;
	*/
	
	/* P4726 【模板】多项式指数函数（多项式 exp）
		int n = read() ;
		Poly A(n) ;
		for(int i = 0 ; i < n ; i ++) A[i] = read() ;
		Poly B = Exp(A , n) ; 
		for(int i = 0 ; i < n ; i ++) write(B[i] , ' ') ;
	*/
	
	/* P5245 【模板】多项式快速幂
		int n = read() ;
		string k ;  cin >> k ;
		Poly A(n) ;
		for(int i = 0 ; i < n ; i ++) A[i] = read() ;
		Poly B = qmi_ntt(A , k , n) ;
		for(int i = 0 ; i < n ; i ++) write(B[i] , ' ') ;
	*/
}
罚时大师月色
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
最小二乘法拟合多项式原理以及c++实现.docx
01-14
void EMatrix(vector<double> Vx, vector<double> Vy, int n, int ex, double coefficient[]) { // ... } void CalEquation(int exp, double coefficient[]) { // ... } double F(double c[], int l, int m) { ...
河流水质模型求解的Chebyshev 正交多项式方法
01-15
<p>提出一种考虑弥散时St reeter2Phelp s 一维稳态河流水质模型Chebyshev 正交多项式的近似解法. 通过对稳<br> 态河流水质模型的非线性高阶微分方程式分析, 采用Chebyshev 正交多项式对各阶微分和弥散系数D 进行...
【学习笔记】多项式全家桶（包含全套证明）
热门推荐
繁凡さん的博客
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.多项式反...
多项式板子
weixin_34138377的博客
03-22 74
咱的多项式板子，从\(NTT\)到反三角函数应有尽有 常数应该比较小，你谷上的板子提交里一般能跑到第一页的 似乎我就没见过几个人现在还在用数组写多项式的了…… //minamoto #include<bits/stdc++.h> #define R register #define fp(i,a,b) for(R int i=(a),I=(b)+1;i<I;++i) #defin...
【全家福】多项式的各种板子
weixin_30291791的博客
04-19 112
写完帕秋莉的超级多项式于是正好贴个模板大汇总（带优化的那种...） //by Judge #include<bits/stdc++.h> #define Rg register #define fp(i,a,b) for(Rg int i=(a),I=(b)+1;i<I;++i) #define fd(i,a,b) for(Rg int i=(a),I=(b)-1;i>I;...
巨型多项式板子集合...
weixin_30426879的博客
04-11 96
300+行完成几乎所有多项式操作... (不得不说指针真是个好东西) #include<bits/stdc++.h> #define Set(a,b) memset(a,b,sizeof(a)) #define Clear(a,_begin_,_end_) for(int i=_begin_;i<_end_;++i) a[i]=0 #define Input_Array(a,_b...
多项式拟合_多项式拟合_VBa_
10-01
在Excel VBA（Visual Basic for Applications）环境中，我们可以利用VBA强大的编程能力来实现复杂的数学运算，例如多项式拟合。多项式拟合是一种通过拟合数据点来创建一个多项式函数的过程，常用于数据分析、预测和...
一元多项式相加Java版本
最新发布
10-11
在Java编程语言中，一元多项式相加是一项基本的数学计算操作，它涉及到数组、循环和算术运算。这个项目提供了一个简单的实现，适用于初学者理解和实践Java编程。我们将会详细探讨`ToolsJava`和`TestMain`两个Java...
Poly.rar_C++多项式_多项式
09-14
std::vector<Term> terms; public: // 构造函数、析构函数等... }; ``` 为了实现多项式的加法，我们需要遍历两个多项式的项，对于相同的指数进行相加，不同的指数则分别保留。这可以通过重载`+`运算符来实现。 `...
多项式开方板子
include / Hdhd
05-07 246
注意常数项要手动开方 void My_Sqrt(int a[],int Sqrt[],int toap) { if(toap==1){Sqrt[0]=Get_Sqrt(a[0]);return;} My_Sqrt(a,Sqrt,toap&gt;&gt;1); fill(Sqrt+(toap&gt;&gt;1),Sqrt+(toap&lt;&lt;1),0); co...
多项式板子·新
weixin_30750335的博客
02-19 94
upd于2.19:拉格朗日反演、多项式倍增快速幂 (好像是还差一个多项式取模...)算了不写了 注意本板子使用过程中：每个函数传的len一定要保证是2的倍数，并且传递的数组需要保证他有多于2*len的空间 每个函数传进来的指针保证[0,len)有值，[len,2*len)有定义 注意new出来的内存一定要清零，不然肯定会炸 #include <cstdio> #include <...
多项式总结&多项式板子
weixin_30256901的博客
06-12 192
多项式总结&多项式板子 三角/反三角是不可能放的（也不可能真香的 多项式乘法（DFT,FFT,NTT,MTT） 背板子 前置知识：泰勒展开 如果\(f(x)\)在\(x_0\)处存在\(n\)阶导，那么 \[f(x)=\sum_{i=0}^{\infty}\frac{f^i(x_0)}{i!}(x-x_0)^i\] 称作\(f(x)\)在\(x_0\)处的泰勒展开。 前置知识：牛顿迭代 ...
多项式全家桶板子
a_forever_dream的博客
05-22 386
短的 int inv[maxn]; int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;} #define INV(x) ksm(x,mod-2) struct NTT{ vector<int> w[30];NTT(){ inv[1]=1;for(int i=2;i<=maxn-10;i++)inv[i]=1ll*(mod-mod/i
【模板】vector<int> 封装模意义下的多项式操作
zxyoi_dreamer的博客（不定期诈尸）
03-14 452
就是放一个自己用的版子，没有讲解。 用vector主要是因为好写且好调。 能在考场上用数组指针码出多项式的所有操作的“神仙”还是留给别人当吧 反正我没见过谁真的为了跑的比vector快，去码了6,7K的操作，我见过的最长的是11K的多项式操作。 写那么优秀有什么屁用呢？反正你考场上也写不完，其次所有用数组写的可读性比vector低一个档次，不好调试啊。而且开了O2可能还没有vector快。。。 #...
【多项式全家桶vector板】可能以后常用的板库
ygmjsjdboy的博客
08-04 284
LOJ菜题 没有标题 非常亲民的板子，只有7K不到（包括火车头，vector不加O2就凉了） 以后可以来调调常数 #pragma GCC optimize(3) #pragma GCC target("avx") #pragma GCC optimize("Ofast") #pragma GCC optimize("inline") #pragma GCC optimize("-fgcse") #pragma GCC optimize("-fgcse-lm") #pragma GCC optimize("-
多元多项式回归 C++带类详细完整实现及案例 变量形式为：vector<vector<double>> X vector<double> y 带有正确的距离计算
06-08
PolynomialRegression(vector<vector<double>> X_, vector<double> y_, int k_) : X(X_), y(y_), n(X_[0].size()), m(X_.size()), k(k_) {} void fit() { // 构造多项式自变量 for (int i = 0; i < m; i++) { ...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值