【知识总结】多项式全家桶(一)(NTT、加减乘除和求逆)
我这种数学一窍不通的菜鸡终于开始学多项式全家桶了……
必须要会的前置技能:FFT(不会?戳我:【知识总结】快速傅里叶变换(FFT))
以下无特殊说明的情况下,多项式的长度指多项式最高次项的次数加\(1\)
一、NTT
跟FFT功能差不多,只是把复数域变成了模域(计算复数系数多项式相乘变成计算在模意义下整数系数多项式相乘)。你看FFT里的单位圆是循环的,模一个质数也是循环的嘛qwq。\(n\)次单位根\(w_n\)怎么搞?看这里:【BZOJ3328】PYXFIB(数学)(内含相关证明。只看与原根和单位根相关的内容即可。)
注意裸的NTT要求模数\(p\)存在原根并且\(p-1\)是\(2\)的若干次幂的倍数(这个幂要大于多项式次数\(n\))。于是通常就会用著名的NTT模数:\(998244353=2^{23}\times 7\times 17+1\)。
节约篇幅,代码先不放了。后面所有代码里都有NTT模板……
二、多项式求逆
对于\(n\)次多项式\(A\),如果有多项式\(B\)满足\(AB\equiv 1 \mod x^{n+1}\),则称\(B\)是\(A\)在模\(x^{n+1}\)意义下的逆元(和整数逆元差不多)。通常采用倍增的方法求逆元。通常都会规定多项式系数在模\(p\)的意义下。
首先,\(A\)在模\(x\)的意义下就只有一个常数项,所以此时的逆元\(B\)也只有一个常数项,就是\(A\)的常数项模\(p\)的逆元。
如果我们知道\(B_0\)是\(A\)在模\(x^{\lceil\frac{n}{2}\rceil}\)意义下的逆元,现在要求\(B\)是\(A\)在模\(x^n\)意义下的逆元。根据题设,显然有:
\]
很明显,\(AB\)的\(1\)到\(n-1\)次项系数全是\(0\),所以模一个\(x\)的低于\(n\)次幂也一定是\(1\)。所以
\]
那么
\]
两边和模数同时平方:
\]
两边同时乘\(A\),得到(别忘了\(AB=1\mod x^n\)):
\]
然后移项,得到:
\]
照着这个式子递归算就行了。
由于后面带余除法的代码包含求逆,所以代码同样略去……
三、加减乘除
加减法:直接每项对应相加减。
乘法:这就是NTT的目的啊喂!
除法:如果不是带余除法直接乘逆元。下面着重介绍带余除法。
已知\(n-1\)次多项式\(F\)和\(m-1\)次多项式\(G\),求\(n-m\)次多项式\(Q\)和多项式\(R\)(\(R\)的次数小于\(m-1\)),满足:
\]
很明显,主要的难点在于式子里有个叫做\(R\)的嘴子(兔崽子Tzz)。如果能把它搞掉该多好……
注意到\(R\)的次数小于\(m-1\),那么我们把它翻转,末尾补\(0\),是不是就可以把它模成\(0\)了?定义\(\mathrm{Tzz}_{A,n}\)表示把\(A\)视作一个长为\(n\)的多项式(高次项补\(0\))后翻转的结果。即\(\mathrm{Tzz}_{A,n}(x)=x^{n-1}A(\frac{1}{x})=\sum\limits_{i=0}^{n-1}a_ix^{n-i-1}\)
给\(F=QG+R\)的每个多项式都代入同一个数,这个多项式也一定是成立的。所以:
\]
两边同乘\(x^{n-1}\),得到:
\]
即
\]
现在\(\mathrm{Tzz}_{R,n}\)的最高次项是\(n-1\),但是从常数项到\(n-m\)次项全是\(0\)(因为\(R\)的长度最多就是\(m-1\))。所以现在如果模\(n-m+1\),那么\(\mathrm{Tzz}_{R,n}\)就是\(0\)了,而\(\mathrm{Tzz}_{Q,n-m+1}\)因为最高次是\(n-m\)所以不会受到影响。
于是用\(\mathrm{Tzz}_{F,n}\)乘上\(\mathrm{Tzz}_{G,m}\)的逆元就是\(\mathrm{Tzz}_{Q,n-m+1}\),翻回去就能得到\(Q\)。
最后把\(Q\)代进原式,乘一乘减一减就能算出\(R\)。
所以这样为什么是对的?(以下“低次项”指翻转后的前\(n-m\)项,“高次项”指翻转后的后\(m\)项)首先在模\(x^{n-m+1}\)意义下肯定能保证低次项是对的(即\(\mathrm{Tzz}{F,n}\)与\(\mathrm{Tzz}_{G,m}\mathrm{Tzz}_{Q,n-m+1}\)的前\(n-m\)项相等)。至于高次项,反正有\(\mathrm{Tzz}_{R,n}\)来补锅,所以即使不对也没关系。
完结撒花。
下一篇:【知识总结】多项式全家桶(二)(ln和exp)
代码:洛谷4512
注意NTT的数组一定要保证多余的元素全部是\(0\)。
代码开头的#undef是防机惨护身符。
(我脑子有病啊求原根全是手写的……
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>
#undef i
#undef j
#undef k
#undef min
#undef max
#undef swap
#undef sort
#undef for
#undef while
#undef if
#undef true
#undef false
#undef printf
#undef scanf
#undef getchar
#undef putchar
#define _ 0
using namespace std;
namespace zyt
{
template<typename T>
inline bool read(T &x)
{
char c;
bool f = false;
x = 0;
do
c = getchar();
while (c != EOF && c != '-' && !isdigit(c));
if (c == EOF)
return false;
if (c == '-')
f = true, c = getchar();
do
x = x * 10 + c - '0', c = getchar();
while (isdigit(c));
if (f)
x = -x;
return true;
}
template<typename T>
inline void write(T x)
{
static char buf[20];
char *pos = buf;
if (x < 0)
putchar('-'), x = -x;
do
*pos++ = x % 10 + '0';
while (x /= 10);
while (pos > buf)
putchar(*--pos);
}
typedef long long ll;
const int N = 1e5 + 10, LEN = (N << 2), p = 998244353;
namespace Polynomial
{
inline int power(int a, int b)
{
a %= p, b %= p - 1;
int ans = 1;
while (b)
{
if (b & 1)
ans = (ll)ans * a % p;
a = (ll)a * a % p;
b >>= 1;
}
return ans;
}
inline int inv(const int a)
{
return power(a, p - 2);
}
namespace Primitive_Root
{
pair<int, int> prime[20];
int cnt;
void get_prime(int n)
{
cnt = 0;
for (int i = 2; i * i <= n; i++)
{
if (n % i == 0)
prime[cnt++] = make_pair(i, 0);
while (n % i == 0)
++prime[cnt - 1].second, n /= i;
}
}
int get_g(const int n)
{
get_prime(n - 1);
for (int i = 2; i < n; i++)
{
bool flag = true;
for (int j = 0; j < cnt && flag; j++)
flag &= (power(i, (n - 1) / prime[j].first) != 1);
if (flag)
return i;
}
return -1;
}
}
int omega[LEN], winv[LEN], rev[LEN];
void init(const int n, const int lg2)
{
static int g = 0;
if (!g)
g = Primitive_Root::get_g(p);
int w = power(g, (p - 1) / n), wi = inv(w);
omega[0] = winv[0] = 1;
for (int i = 1; i < n; i++)
{
omega[i] = (ll)omega[i - 1] * w % p;
winv[i] = (ll)winv[i - 1] * wi % p;
}
for (int i = 0; i < n; i++)
rev[i] = ((rev[i >> 1] >> 1) | ((i & 1) << (lg2 - 1)));
}
void ntt(int *a, const int *w, const int n)
{
for (int i = 0; i < n; i++)
if (i < rev[i])
swap(a[i], a[rev[i]]);
for (int l = 1; l < n; l <<= 1)
for (int i = 0; i < n; i += (l << 1))
for (int k = 0; k < l; k++)
{
int tmp = (a[i + k] - (ll)w[n / (l << 1) * k] * a[i + l + k] % p + p) % p;
a[i + k] = (a[i + k] + (ll)w[n / (l << 1) * k] * a[i + l + k] % p) % p;
a[i + l + k] = tmp;
}
}
void reverse(int *a, const int n)
{
static int tmp[LEN];
memcpy(tmp, a, sizeof(int[n]));
for (int i = 0; i < n; i++)
a[i] = tmp[n - i - 1];
}
inline void plus(const int *a, const int *b, int *c, const int n)
{
for (int i = 0; i < n; i++)
c[i] = (a[i] + b[i]) % p;
}
inline void minus(const int *a, const int *b, int *c, const int n)
{
for (int i = 0; i < n; i++)
c[i] = (a[i] - b[i] + p) % p;
}
void _inv(const int *a, int *b, const int n)
{
if (n == 1)
b[0] = inv(a[0]);
else
{
static int tmp[LEN];
_inv(a, b, (n + 1) >> 1);
int m = 1, lg2 = 0;
while (m < (n << 1) - 1)
m <<= 1, ++lg2;
memcpy(tmp, a, sizeof(int[n]));
memset(tmp + n, 0, sizeof(int[m - n]));
memset(b + ((n + 1) >> 1), 0, sizeof(int[m - ((n + 1) >> 1)]));
init(m, lg2);
ntt(tmp, omega, m);
ntt(b, omega, m);
for (int i = 0; i < m; i++)
b[i] = (b[i] * 2LL % p - (ll)tmp[i] * b[i] % p * b[i] % p + p) % p;
ntt(b, winv, m);
int invm = inv(m);
for (int i = 0; i < m; i++)
b[i] = (ll)b[i] * invm % p;
memset(b + n, 0, sizeof(int[m - n]));
}
}
void inv(const int *a, int *b, const int n)
{
static int tmp[LEN];
memcpy(tmp, a, sizeof(int[n]));
_inv(tmp, b, n);
}
void mul(const int *a, const int *b, int *c, const int n)
{
int m = 1, lg2 = 0;
while (m < (n << 1))
m <<= 1, ++lg2;
static int x[LEN], y[LEN];
memcpy(x, a, sizeof(int[n]));
memset(x + n, 0, sizeof(int[m - n]));
memcpy(y, b, sizeof(int[n]));
memset(y + n, 0, sizeof(int[m - n]));
init(m, lg2);
ntt(x, omega, m);
ntt(y, omega, m);
for (int i = 0; i < m; i++)
x[i] = (ll)x[i] * y[i] % p;
ntt(x, winv, m);
int invm = inv(m);
for (int i = 0; i < m; i++)
x[i] = (ll)x[i] * invm % p;
memcpy(c, x, sizeof(int[n]));
}
void div(const int *_F, const int *_G, int *_Q, int *_R, const int n, const int m)
{
static int F[LEN], G[LEN], invG[LEN], Q[LEN], R[LEN];
memcpy(F, _F, sizeof(int[n]));
memcpy(G, _G, sizeof(int[m]));
reverse(F, n), reverse(G, m);
if (m < n - m + 1)
memset(G + m, 0, sizeof(int[n - m + 1 - m]));
inv(G, invG, n - m + 1);
mul(F, invG, Q, n - m + 1);
reverse(F, n), reverse(G, m), reverse(Q, n - m + 1);
mul(G, Q, G, n);
minus(F, G, R, n);
memcpy(_Q, Q, sizeof(int[n - m + 1]));
memcpy(_R, R, sizeof(int[m]));
}
}
int F[LEN], G[LEN], Q[LEN], R[LEN];
int work()
{
int n, m;
read(n), read(m);
++n, ++m;
for (int i = 0; i < n; i++)
read(F[i]);
for (int i = 0; i < m; i++)
read(G[i]);
Polynomial::div(F, G, Q, R, n, m);
for (int i = 0; i < n - m + 1; i++)
write(Q[i]), putchar(' ');
putchar('\n');
for (int i = 0; i < m - 1; i++)
write(R[i]), putchar(' ');
return (0^_^0);
}
}
int main()
{
return zyt::work();
}
【知识总结】多项式全家桶(一)(NTT、加减乘除和求逆)的更多相关文章
- 【知识总结】多项式全家桶(三)(任意模数NTT)
经过两个月的咕咕,"多项式全家桶" 系列终于迎来了第三期--(雾) 上一篇:[知识总结]多项式全家桶(二)(ln和exp) 先膜拜(伏地膜)大恐龙的博客:任意模数 NTT (在页面 ...
- 【知识总结】多项式全家桶(二)(ln和exp)
上一篇:[知识总结]多项式全家桶(一)(NTT.加减乘除和求逆) 一.对数函数\(\ln(A)\) 求一个多项式\(B(x)\),满足\(B(x)=\ln(A(x))\). 这里需要一些最基本的微积分 ...
- [模板]多项式全家桶小记(求逆,开根,ln,exp)
前言 这里的全家桶目前只包括了\(ln,exp,sqrt\).还有一些类似于带余数模,快速幂之类用的比较少的有时间再更,\(NTT\)这种前置知识这里不多说. 还有一些基本的导数和微积分内容要了解,建 ...
- Solution -「LOJ #150」挑战多项式 ||「模板」多项式全家桶
\(\mathcal{Description}\) Link. 给定 \(n\) 次多项式 \(F(x)\),在模 \(998244353\) 意义下求 \[G(x)\equiv\left\{ ...
- loj#6363. 「地底蔷薇」(拉格朗日反演+多项式全家桶)
题面 传送门 题解 肝了一个下午--我老是忘了拉格朗日反演计算的时候多项式要除以一个\(x\)--结果看它推倒简直一脸懵逼-- 做这题首先你得知道拉格朗日反演是个什么东西->这里 请坐稳,接下来 ...
- bzoj3684: 大朋友和多叉树(拉格朗日反演+多项式全家桶)
题面 传送门 题解 首先你得知道什么是拉格朗日反演->这里 我们列出树的个数的生成函数 \[T(x)=x+\prod_{i\in D}T^i(x)\] \[T(x)-\prod_{i\in D} ...
- [模板] 多项式: 乘法/求逆/分治fft/微积分/ln/exp/幂
多项式 代码 const int nsz=(int)4e5+50; const ll nmod=998244353,g=3,ginv=332748118ll; //basic math ll qp(l ...
- IP 基础知识全家桶,45 张图一套带走
前言 前段时间,有读者希望我写一篇关于 IP 分类地址.子网划分等的文章,他反馈常常混淆,摸不着头脑. 那么,说来就来!而且要盘就盘全一点,顺便挑战下小林的图解功力,所以就来个 IP 基础知识全家桶. ...
- Vue全家桶之一Vue(基础知识篇)
全家桶:Vue本身.状态管理.路由. 异步组件:
随机推荐
- apache 添加虚拟机
<VirtualHost *:80> DocumentRoot "E:/UPUPW_AP7.0/htdocs/xd.local/public" ServerName a ...
- Python基础-获取当前目录,上级目录,上上级目录
import os print '***获取当前目录***' print os.getcwd() print os.path.abspath(os.path.dirname(__file__)) pr ...
- (一)U-Boot启动过程--详细版的完全分析
博客地址:http://blog.csdn.net/hare_lee/article/details/6916325
- BZOJ 4385 洛谷3594 POI2015 WIL-Wilcze doły
[题解] 手残写错调了好久QAQ...... 洛谷的数据似乎比较水.. n个正整数!!这很重要 这道题是个类似two pointer的思想,外加一个单调队列维护当前区间内长度为d的子序列中元素之和的最 ...
- 【Codeforces 279C】Ladder
[链接] 我是链接,点我呀:) [题意] 题意 [题解] 设pre[i]表示i往前一直递增能递增多远 设aft[i]表示i往后一直递增能递增多远 如果aft[l]+pre[r]>=(r-l+1) ...
- HDU 1059 多重背包问题
问题大意: 有价值1-6的六种物品,分别规定其数目,问是否存在一种方法能使这些物品不拆分就能平均分给两个人 #include <cstdio> #include <cstring&g ...
- 由MTK平台 mtkfb 设备注册疑问引发的知识延伸--ARM Device Tree
问题: 在kernel-3.10\drivers\misc\mediatek\videox\mt6735\mtkfb.c里面int __init mtkfb_init(void) 有看到 platfo ...
- [bzoj 2705][SDOI2012]Longge的问题(数学)
题目:http://www.lydsy.com:808/JudgeOnline/problem.php?id=2705 分析: 设k为n的因数 设f[k]为gcd(x,n)==k的x的个数,容易知道a ...
- 联想YOGA3一键恢复系统教程
- vijos - P1302连续自然数和 (公式推导 + python)
P1302连续自然数和 Accepted 标签:[显示标签] 描写叙述 对一个给定的自然数M,求出所有的连续的自然数段(连续个数大于1).这些连续的自然数段中的所有数之和为M. 样例:1998+199 ...