P5394 【模板】下降幂多项式乘法

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

题目不好复制

要吸吸氧才能过。
这种写法比第二种写法少一次ntt。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#define ll long long
#define llu unsigned ll
#define int ll
using namespace std;
const int maxn=8e5+1000;
const int p=998244353;
const int g=3;
int fi[maxn];
int a[maxn],b[maxn],c[maxn];
int fac[maxn],inv[maxn];

int mypow(int a,int b)
{
    if(b<0) return mypow(mypow(a,p-2),-b);
    int ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%p;
        a=a*a%p;
        b>>=1;
    }
    return ans%p;
}

void init(int n)
{
    fac[0]=1;
    for(int i=1;i<=n;i++)
        fac[i]=fac[i-1]*i%p;
    inv[n]=mypow(fac[n],p-2);
    for(int i=n-1;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%p;
}

void ntt(int *x,int len,int f)
{
    for(int i=0;i<len;i++)
        if(i<fi[i]) swap(x[i],x[fi[i]]);

    for(int i=1;i<len;i<<=1)
    {
        int r=i<<1;
        int wn=mypow(g,f*(p-1)/r);
        for(int j=0;j<len;j+=r)
        {
            int w=1;
            for(int k=0;k<i;k++)
            {
                int xx=x[j+k],yy=w*x[j+i+k]%p;
                x[j+k]=(xx+yy)%p;
                x[j+i+k]=((xx-yy)%p+p)%p;
                w=w*wn%p;
            }
        }
    }
    if(f==-1)
    {
        int invn=mypow(len,p-2);
        for(int i=0;i<len;i++)
            x[i]=x[i]*invn%p;
    }
}


signed main(void)
{
    int n,m;
    scanf("%lld%lld",&n,&m);
    for(int i=0;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=0;i<=m;i++)
        scanf("%lld",&b[i]);

    int now=n+m;
    n=now,m=now;
    int len=1,cnt=0;
    while(len<=n+m) len<<=1,cnt++;
    for(int i=0;i<len;i++)
        fi[i]=((fi[i>>1]>>1)|((i&1)<<(cnt-1)));

    init(now);
    for(int i=0;i<=now;i++)
        c[i]=inv[i];
    ntt(a,len,1);
    ntt(b,len,1);
    ntt(c,len,1);
    for(int i=0;i<len;i++)
        a[i]=a[i]*c[i]%p,b[i]=b[i]*c[i]%p;
    ntt(a,len,-1);
    ntt(b,len,-1);
    for(int i=0;i<=now;i++)
        a[i]=a[i]*b[i]%p*fac[i]%p;
    for(int i=now+1;i<len;i++)
        a[i]=0,c[i]=0;


    for(int i=0;i<=now;i++)
    {
        if(i&1) c[i]=-inv[i]+p;
        else c[i]=inv[i];
    }
    ntt(a,len,1);
    ntt(c,len,1);
    for(int i=0;i<len;i++)
        a[i]=a[i]*c[i]%p;
    ntt(a,len,-1);


    for(int i=0;i<=now;i++)
        printf("%lld ",a[i]);
    putchar('\n');
    return 0;

}

只是比上面那种写法多一次ntt而已。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#define ll long long
#define llu unsigned ll
#define int ll
using namespace std;
const int maxn=8e5+1000;
const int p=998244353;
const int g=3;
int fi[maxn];
int a[maxn],b[maxn],c[maxn];
int fac[maxn],inv[maxn];

int mypow(int a,int b)
{
    if(b<0) return mypow(mypow(a,p-2),-b);
    int ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%p;
        a=a*a%p;
        b>>=1;
    }
    return ans%p;
}

void init(int n)
{
    fac[0]=1;
    for(int i=1;i<=n;i++)
        fac[i]=fac[i-1]*i%p;
    inv[n]=mypow(fac[n],p-2);
    for(int i=n-1;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%p;
}

void ntt(int *x,int len,int f)
{
    for(int i=0;i<len;i++)
        if(i<fi[i]) swap(x[i],x[fi[i]]);

    for(int i=1;i<len;i<<=1)
    {
        int r=i<<1;
        int wn=mypow(g,f*(p-1)/r);
        for(int j=0;j<len;j+=r)
        {
            int w=1;
            for(int k=0;k<i;k++)
            {
                int xx=x[j+k],yy=w*x[j+i+k]%p;
                x[j+k]=(xx+yy)%p;
                x[j+i+k]=((xx-yy)%p+p)%p;
                w=w*wn%p;
            }
        }
    }
    if(f==-1)
    {
        int invn=mypow(len,p-2);
        for(int i=0;i<len;i++)
            x[i]=x[i]*invn%p;
    }
}

void getntt(int *a,int len,int n,int f)
{
    if(f==1)
    {
        for(int i=0;i<=n;i++)
            c[i]=inv[i];
    }
    else
    {
        for(int i=0;i<=n;i++)
        {
            if(i&1) c[i]=-inv[i]+p;
            else c[i]=inv[i];
        }
    }

    for(int i=n+1;i<=len;i++)
        c[i]=0;

    ntt(a,len,1);
    ntt(c,len,1);
    for(int i=0;i<len;i++)
        a[i]=a[i]*c[i]%p;
    ntt(a,len,-1);
}

signed main(void)
{
    int n,m;
    scanf("%lld%lld",&n,&m);
    for(int i=0;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=0;i<=m;i++)
        scanf("%lld",&b[i]);

    int now=n+m;
    n=now,m=now;
    int len=1,cnt=0;
    while(len<=n+m) len<<=1,cnt++;
    for(int i=0;i<len;i++)
        fi[i]=((fi[i>>1]>>1)|((i&1)<<(cnt-1)));

    init(now);
    getntt(a,len,now,1);
    getntt(b,len,now,1);
    for(int i=0;i<=now;i++)
        a[i]=a[i]*b[i]%p*fac[i]%p;
    for(int i=now+1;i<len;i++)
        a[i]=0;
    getntt(a,len,now,-1);
    for(int i=0;i<=now;i++)
        printf("%lld ",a[i]);
    putchar('\n');
    return 0;

}

加上快读，再加上预处理原根的阶乘，也没卡过。。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#define ll long long
#define llu unsigned ll
#define int ll
using namespace std;

char buffer[100001],*S,*T;
inline char Get_Char()
{
    if (S==T)
    {
        T=(S=buffer)+fread(buffer,1,100001,stdin);
        if (S==T) return EOF;
    }
    return *S++;
}
inline int read()
{
    char c;int re=0;
    for(c=Get_Char();c<'0'||c>'9';c=Get_Char());
    while(c>='0'&&c<='9') re=re*10+(c-'0'),c=Get_Char();
    return re;
}


const int maxn=8e5+1000;
const int p=998244353;
const int g=3;
int fi[maxn];
int a[maxn],b[maxn],c[maxn];
int fac[maxn],inv[maxn],gg[2][maxn];

int mypow(int a,int b)
{
    if(b<0) return mypow(mypow(a,p-2),-b);
    int ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%p;
        a=a*a%p;
        b>>=1;
    }
    return ans%p;
}

void init(int n,int m)
{
    fac[0]=1;
    for(int i=1;i<=n;i++)
        fac[i]=fac[i-1]*i%p;
    inv[n]=mypow(fac[n],p-2);
    for(int i=n-1;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%p;
    for(int i=1;i<(m<<1);i<<=1)
        gg[0][i]=mypow(g,-(p-1)/i),gg[1][i]=mypow(g,(p-1)/i);
}

void ntt(int *x,int len,int f)
{
    for(int i=0;i<len;i++)
        if(i<fi[i]) swap(x[i],x[fi[i]]);

    for(int i=1;i<len;i<<=1)
    {
        int r=i<<1;
        int wn=gg[f][r];
        for(int j=0;j<len;j+=r)
        {
            int w=1;
            for(int k=0;k<i;k++)
            {
                int xx=x[j+k],yy=w*x[j+i+k]%p;
                x[j+k]=(xx+yy)%p;
                x[j+i+k]=((xx-yy)%p+p)%p;
                w=w*wn%p;
            }
        }
    }
    if(f==0)
    {
        int invn=mypow(len,p-2);
        for(int i=0;i<len;i++)
            x[i]=x[i]*invn%p;
    }
}


signed main(void)
{
    int n,m;
    n=read(),m=read();
    for(int i=0;i<=n;i++)
        a[i]=read();
    for(int i=0;i<=m;i++)
        b[i]=read();

    int now=n+m;
    n=now,m=now;
    int len=1,cnt=0;
    while(len<=n+m) len<<=1,cnt++;
    for(int i=0;i<len;i++)
        fi[i]=((fi[i>>1]>>1)|((i&1)<<(cnt-1)));

    init(now,len);
    for(int i=0;i<=now;i++)
        c[i]=inv[i];
    ntt(a,len,1);
    ntt(b,len,1);
    ntt(c,len,1);
    for(int i=0;i<len;i++)
        a[i]=a[i]*c[i]%p,b[i]=b[i]*c[i]%p;
    ntt(a,len,0);
    ntt(b,len,0);
    for(int i=0;i<=now;i++)
        a[i]=a[i]*b[i]%p*fac[i]%p;
    for(int i=now+1;i<len;i++)
        a[i]=0,c[i]=0;


    for(int i=0;i<=now;i++)
    {
        if(i&1) c[i]=-inv[i]+p;
        else c[i]=inv[i];
    }
    ntt(a,len,1);
    ntt(c,len,1);
    for(int i=0;i<len;i++)
        a[i]=a[i]*c[i]%p;
    ntt(a,len,0);


    for(int i=0;i<=now;i++)
        printf("%lld ",a[i]);
    putchar('\n');
    return 0;

}
_Shmily
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
洛谷P5394 【模板】下降幂多项式乘法（生成函数）
苟为蒟蒻又何妨
08-02 277
传送门 考虑求出如下两个多项式的乘积： A(x)=∑i=0naixi‾,B(x)=∑i=0mbixi‾A(x)=\sum_{i=0}^na_ix^{\underline i},B(x)=\sum_{i=0}^mb_ix^{\underline i}A(x)=∑i=0n​ai​xi​,B(x)=∑i=0m​bi​xi​ 然后考虑把它们转成点值形式然后乘起来再转回去。 设现在F(x)=∑i=0nfix...
luogu P5394 【模板】下降幂多项式乘法
lahlah的博客
12-21 261
https://www.luogu.com.cn/problem/P5394 之前一直以为是什么阴间东西，没有碰，现在菜知道原来是个挺naive的东西
【模板】【洛谷P5394】下降幂多项式乘法（NTT）
zxyoi_dreamer的博客（不定期诈尸）
08-13 197
传送门 题解： 对于下降幂多项式f(x)f(x)f(x)考虑其点值生成函数： F(x)=∑i=0∞f(i)i!⋅xiF(x)=\sum_{i=0}^\infty\frac{f(i)}{i!}\cdot x^iF(x)=i=0∑∞​i!f(i)​⋅xi 考虑一个下降幂单项式xn‾x^{\underline{n}}xn​的点值生成函数：∑i=n∞i!(n−i)!i!xi=xnexp⁡(x)\sum_...
【洛谷P5394】【模板】—下降幂多项式乘法（指数级生成函数）
Stargazer的博客
08-03 419
传送门 首先考虑对于xn‾x^{\underline n}xn​构建EGFEGFEGF xn‾=∑i=n∞in‾i!xi=∑i=n∞1(i−n)!xi=xnexx^{\underline n}=\sum_{i=n}^{\infty}\frac{i^{\underline n}}{i!}x^i=\sum_{i=n}^{\infty}\frac{1}{(i-n)!}x^i=x^ne^xxn​=∑i=n...
【算法讲15：多项式卷积】FFT | NTT
溢流眼泪的博客
02-17 4051
【算法讲15：多项式】多项式的表示法多项式的乘法与卷积多项式乘法FFTFFTFFT单位根离散傅里叶变换 DFTDFTDFTDFTDFTDFT 的详细讨论离散傅里叶逆变换 IDFTIDFTIDFT参考与查阅 多项式的表示法 系数表示：nnn 个系数确定一个 n−1n-1n−1 次多项式 (a0,a1,a2,⋯ ,an−1) (a_0,a_1,a_2,\cdots,a_{n-1}) (a0​,a1​,a2​,⋯,an−1​) 点值表示：nnn 个点确定一个 n−1n-1n−1 刺多项式 (x0,x1,x1,⋯
多项式乘法实现
12-22
数据结构作业，适合初学者，小白，自学者，和大家共享。
MUL.rar_多项式乘法
09-14
在这个"多项式乘法"的实验中，我们聚焦于利用编程技术实现数学中的多项式乘法，并进行结果的排序。这个实验主要涉及以下几个关键知识点： 1. **多项式表示**：在计算机中，多项式通常用数组或链表等数据结构来表示...
Java实现多项式乘法代码实例
08-26
Java实现多项式乘法代码实例 本文介绍了Java实现多项式乘法代码实例的详细过程，包括代码实现和结果输出。多项式乘法是数学中的一种基本运算，它广泛应用于计算机科学、工程技术、物理学等领域。Java是目前最流行的...
多项式乘法（java源代码）
05-21
在计算机科学领域，尤其是数值计算和算法设计中，多项式乘法是一项基本操作。这个Java源代码实现的目的是为了高效地处理两个多项式之间的乘法运算。多项式通常表示为一系列系数与变量的乘积之和，例如 \(P(x) = a_nx...
算法文档无代码多项式乘法算法文档无代码多项式乘法
最新发布
04-14
算法文档无代码多项式乘法算法文档无代码多项式乘法提取方式是百度网盘分享地址
质数（素数）、约数
_Griefs
08-24 1331
一、质数： 若一个正整数无法被除了1和它自身之外的任何自然数整除，则称该数为质数（或素数），否则称该正整数为合数。 1不是素数，2是最小的素数。 在整个自然数集合中，质数的数量不多，分布比较稀疏，对于一个最够大的整数N，不超过N的质数大约有N/lnN个，即每lnN个数大约有1个质数。 二、素数表的获取——Eratosthenes筛法： 小于x*x的x的倍数在扫描更小的数时就已经被标记过了。对于某一...
求a的b次方对mod取模：
_Griefs
08-24 801
一、 分治： #include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #include<string> #include<cmath> #define ll long long using namespace std; const in...
P6091 【模板】原根
_Griefs
08-24 652
求全部原根的板子： #include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #include<cmath> #include<string> #include<queue> #define ll long long #def...
中国剩余定理：
_Griefs
08-24 455
一、标版： 题目描述 现有两组数字，每组k个，第一组中的数字分别为：a1，a2，…，ak表示，第二组中的数字分别用b1，b2，…，bk表示。其中第二组中的数字是两两互素的。求最小的非负整数n，满足对于任意的i，n - ai能被bi整除。 输入输出格式 输入格式： 输入数据的第一行是一个整数k，（1 ≤ k ≤ 10）。接下来有两行，第一行是：a1，a2，…，ak，第二行是b1，b2，…，bk 输出...
多项式求逆
_Griefs
08-24 451
一、模板 P4238 【模板】多项式求逆： 题目背景 模板题，无背景 题目描述 给定一个多项式 F(x) ，请求出一个多项式 G(x)， 满足F(x)∗G(x)≡1(modxn)。系数对 998244353 取模。 输入格式 首先输入一个整数 n， 表示输入多项式的项数。 接着输入 n 个整数，第 i 个整数 ai代表F(x) 次数为 i−1 的项的系数。 输出格式 输出 n 个数字，第 i 个整...
欧几里得、扩展欧几里得、同余
_Griefs
08-24 424
一、同余： 1。若整数a和整数b除以正整数m的余数相等，则称a，b模m同余，记作a≡b（mod m）。 2。费马小定理：若p是质数，则对于任意整数a，有ap ≡a（mod p）。 3。欧拉定理：若正整数a，n互质，则 aφ(n)≡1(mod n) ,其中φ(n)是欧拉函数。 4。若正整数a，n互质，则对于任意正整数b，有ab≡ab mod φ(n)(mod n)。 很多问题要求我们对一个质数p取...
NTT
_Griefs
08-24 395
一、原根： 例题：51nod 1135 原根： 设m是正整数，a是整数，若a模m的阶等于φ(m)，则称a为模m的一个原根。（其中φ(m)表示m的欧拉函数） 给出1个质数P，找出P最小的原根。 输入 输入1个质数P(3 <= P <= 10^9) 输出 输出P最小的原根。 输入样例 3 输出样例 2 #include<iostream> #include<cstdio&...
pell方程
_Griefs
08-24 386
一、连分数法求解最小正整数解。 自己看着公式敲的。。没有题目检验，但是暴力对拍了一会，应该是对的吧。。。 #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cmath> #include<queue> #incl...
python 多项式乘法
09-29
Python多项式乘法可以通过使用嵌套的循环来实现。具体步骤如下： 1. 首先，我们需要定义两个多项式。假设多项式P(x)的系数为[a0, a1, a2, ..., an]，多项式Q(x)的系数为[b0, b1, b2, ..., bm]。 2. 接下来，我们将...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值