多项式各种操作

最新推荐文章于 2019-05-06 18:31:00 发布

star-city

最新推荐文章于 2019-05-06 18:31:00 发布

阅读量430

点赞数

分类专栏： NTT 多项式

本文链接：https://blog.csdn.net/star_city_7/article/details/81143378

版权

这篇博客总结了多项式的一些关键操作，包括使用三模数NTT进行乘法，逆元的计算，平方根的求解，以及除法和取模的处理。通过介绍相关算法和代码实现，阐述了这些操作在多项式计算中的应用。

摘要由CSDN通过智能技术生成

前些天学了一堆多项式的算法，今天总结一下。
##乘法
朴素的NTT就不说了，主要说一下三模数NTT。
三模数NTT，顾名思义，就是选取三个适合做NTT的模数，然后把它们用CRT合并起来得到的答案再去对我们要求的模数取模。
为了方便，这三个模数分别是998244353, 1004535809和469762049。它们的原根都是3，且它们减去1的值都有超过20个2的因子。
~~如果你觉得能用int128的话就请忽略下面所有的表述~~
但是，在用CRT合并的时候，我们遇到一个麻烦：这三个模数的乘积太大了。我们令三个模数分别为p1, p2, p3，对三个模数取模得到的答案分别为a1, a2, a3，不做取模的原本的答案为Ans，则有：
$\equiv a1~(mod~p1)$
$\equiv a2~(mod~p2)$
$\equiv a3~(mod~p3)$
先用CRT合并前两个模数的答案，得到
$\equiv A~(mod~M)$
$\equiv a3~(mod~p3)$
设
$A n s = k M + A$
那么
$\equiv a3~(mod~p3)$
即
$\equiv \frac{(a3 - A)}{M}~(mod~p3)$
此时k, M, A我们都已经得到，就直接对我们要求的模数取模就好了，即
$Ans~\%~p = ((k~\%~p) * (M~\%~p) + A)~\%~p$
##Code

inline void exntt(int *a, int *b)
{
   
    int len = 1, bit = 0;
    while (len < n + m) len <<= 1, bit++;
    for (int i = 0; i < len; i++) rev[i] = rev[i >> 1] >> 1 | (i & 1) << bit - 1;
    for (int i = 0; i < 3; i++) {
   
        for (int j = 0; j < len; j++) c[j] = a[j], d[j] = b[j];
        ntt(c, len, 1, p[i]);
        ntt(d, len, 1, p[i]);
        for (int j = 0; j < len; j++) ans[i][j] = 1ll * c[j] * d[j] % p[i];
        ntt(ans[i], len, -1, p[i]