[洛谷P5205]【模板】多项式开根
题目大意:给你$n$项多项式$A(x)$,求出$B(x)$满足$B^2(x)\equiv A(x)\pmod{x^n}$
题解:考虑已经求出$B_0(x)$满足$B_0^2(x)\equiv A(x)\pmod{x^{\lceil\frac n 2\rceil}}$
$$
B(x)-B_0(x)\equiv0\pmod{x^{\lceil\frac n 2\rceil}}\\
B^2(x)−2B(x)B_0(x)+B_0^2(x)≡0\pmod{x^n}\\
A(x)-2B(x)B_0(x)+B_0^2(x)≡0\pmod{x^n}\\
B(x)\equiv\dfrac{A(x)+B_0^2(x)}{2B_0(x)}\pmod{x^n}\\
$$
update:(2019-2-10)
$$
B(x)\equiv\dfrac{A(x)+B_0^2(x)}{2B_0(x)}\pmod{x^n}\\
B(x)\equiv\dfrac{A(x)}{2B_0(x)}+\dfrac{B_0(x)}2\pmod{x^n}\\
$$
发现$\dfrac{B_0(x)}2$只会影响$B(x)$数组的前半部分(即$\pmod{x^{\lceil\frac n2\rceil}}$的部分),但是$B(x)\equiv B_0(x)\pmod{x^{\lceil\frac n2\rceil}}$,所以可以不做考虑,直接把$B_0(x)$拉过来
卡点:求$INV$时注意清空数组,防止因为$B$数组不干净导致出锅
C++ Code:
- #include <algorithm>
- #include <cctype>
- #include <cstdio>
- #define maxn 262144
- const int mod = 998244353, __2 = mod + 1 >> 1;
- namespace std {
- struct istream {
- #define M (1 << 21 | 3)
- char buf[M], *ch = buf - 1;
- inline istream() { fread(buf, 1, M, stdin); }
- inline istream& operator >> (int &x) {
- while (isspace(*++ch));
- for (x = *ch & 15; isdigit(*++ch); ) x = x * 10 + (*ch & 15);
- return *this;
- }
- #undef M
- } cin;
- struct ostream {
- #define M (1 << 21 | 3)
- char buf[M], *ch = buf - 1;
- inline ostream& operator << (int x) {
- if (!x) {*++ch = '0'; return *this;}
- static int S[20], *top; top = S;
- while (x) {*++top = x % 10 ^ 48; x /= 10;}
- for (; top != S; --top) *++ch = *top;
- return *this;
- }
- inline ostream& operator << (const char x) {*++ch = x; return *this;}
- inline ~ostream() { fwrite(buf, 1, ch - buf + 1, stdout); }
- #undef M
- } cout;
- }
- namespace Math {
- inline int pw(int base, int p) {
- static int res;
- for (res = 1; p; p >>= 1, base = static_cast<long long> (base) * base % mod) if (p & 1) res = static_cast<long long> (res) * base % mod;
- return res;
- }
- inline int inv(int x) { return pw(x, mod - 2); }
- }
- inline void reduce(int &x) { x += x >> 31 & mod; }
- inline void clear(register int *l, const int *r) {
- if (l >= r) return ;
- while (l != r) *l++ = 0;
- }
- namespace Poly {
- #define N maxn
- int lim, s, rev[N];
- int Wn[N + 1];
- inline void init(const int n) {
- lim = 1, s = -1; while (lim < n) lim <<= 1, ++s;
- for (register int i = 1; i < lim; ++i) rev[i] = rev[i >> 1] >> 1 | (i & 1) << s;
- const int t = Math::pw(3, (mod - 1) / lim);
- *Wn = 1; for (register int *i = Wn; i != Wn + lim; ++i) *(i + 1) = static_cast<long long> (*i) * t % mod;
- }
- inline void FFT(int *A, const int op = 1) {
- for (register int i = 1; i < lim; ++i) if (i < rev[i]) std::swap(A[i], A[rev[i]]);
- for (register int mid = 1; mid < lim; mid <<= 1) {
- const int t = lim / mid >> 1;
- for (register int i = 0; i < lim; i += mid << 1)
- for (register int j = 0; j < mid; ++j) {
- const int X = A[i + j], Y = static_cast<long long> (A[i + j + mid]) * Wn[t * j] % mod;
- reduce(A[i + j] += Y - mod), reduce(A[i + j + mid] = X - Y);
- }
- }
- if (!op) {
- const int ilim = Math::inv(lim);
- for (register int *i = A; i != A + lim; ++i) *i = static_cast<long long> (*i) * ilim % mod;
- std::reverse(A + 1, A + lim);
- }
- }
- void INV(int *A, int *B, int n) {
- if (n == 1) { *B = Math::inv(*A); return ; }
- const int len = n + 1 >> 1;
- INV(A, B, len); init(len * 3);
- static int C[N], D[N];
- std::copy(A, A + n, C); clear(C + n, C + lim);
- std::copy(B, B + len, D); clear(D + len, D + lim);
- FFT(D), FFT(C);
- for (register int i = 0; i < lim; ++i) D[i] = (2 - static_cast<long long> (D[i]) * C[i] % mod + mod) * D[i] % mod;
- FFT(D, 0); std::copy(D + len, D + n, B + len);
- }
- void SQRT(int *A, int *B, int n) {
- if (n == 1) { *B = 1; return ; }
- static int C[N], D[N];
- const int len = n + 1 >> 1;
- SQRT(A, B, len);
- INV(B, D, n), clear(D + n, D + lim);
- std::copy(A, A + n, C); clear(C + n, C + lim);
- FFT(C), FFT(D);
- for (register int i = 0; i < lim; ++i) D[i] = static_cast<long long> (C[i]) * D[i] % mod * __2 % mod;
- FFT(D, 0); std::copy(D + len, D + n, B + len);
- }
- #undef N
- }
- int n, A[maxn], B[maxn];
- int main() {
- std::cin >> n;
- for (int i = 0; i < n; ++i) std::cin >> A[i];
- Poly::SQRT(A, B, n);
- for (int i = 0; i < n; ++i) std::cout << B[i] << ' ';
- std::cout << '\n';
- return 0;
- }
[洛谷P5205]【模板】多项式开根的更多相关文章
- 洛谷.3803.[模板]多项式乘法(FFT)
题目链接:洛谷.LOJ. FFT相关:快速傅里叶变换(FFT)详解.FFT总结.从多项式乘法到快速傅里叶变换. 5.4 又看了一遍,这个也不错. 2019.3.7 叕看了一遍,推荐这个. #inclu ...
- 洛谷.3803.[模板]多项式乘法(NTT)
题目链接:洛谷.LOJ. 为什么和那些差那么多啊.. 在这里记一下原根 Definition 阶 若\(a,p\)互质,且\(p>1\),我们称使\(a^n\equiv 1\ (mod\ p)\ ...
- 洛谷.4512.[模板]多项式除法(NTT)
题目链接 多项式除法 & 取模 很神奇,记录一下. 只是主要部分,更详细的和其它内容看这吧. 给定一个\(n\)次多项式\(A(x)\)和\(m\)次多项式\(D(x)\),求\(deg(Q) ...
- 洛谷.4238.[模板]多项式求逆(NTT)
题目链接 设多项式\(f(x)\)在模\(x^n\)下的逆元为\(g(x)\) \[f(x)g(x)\equiv 1\ (mod\ x^n)\] \[f(x)g(x)-1\equiv 0\ (mod\ ...
- 洛谷 P4512 [模板] 多项式除法
题目:https://www.luogu.org/problemnew/show/P4512 看博客:https://www.cnblogs.com/owenyu/p/6724611.html htt ...
- 洛谷 P4238 [模板] 多项式求逆
题目:https://www.luogu.org/problemnew/show/P4238 看博客:https://www.cnblogs.com/xiefengze1/p/9107752.html ...
- 洛谷P2293 高精开根
锣鼓2293 写完了放代码 应该没什么思维难度 ———————————————————————————————————————————————————————— python真香 m=input() ...
- P5277 【模板】多项式开根(加强版)(bsgs or Cipolla)
题面 传送门 题解 首先你得会多项式开根->这里 其次你得会解形如 \[x^2\equiv a \pmod{p}\] 的方程 这里有两种方法,一个是\(bsgs\)(这里),还有一种是\(Cip ...
- FFT模板 生成函数 原根 多项式求逆 多项式开根
FFT #include<iostream> #include<cstring> #include<cstdlib> #include<cstdio> ...
- 洛谷P3373 [模板]线段树 2(区间增减.乘 区间求和)
To 洛谷.3373 [模板]线段树2 题目描述 如题,已知一个数列,你需要进行下面两种操作: 1.将某区间每一个数加上x 2.将某区间每一个数乘上x 3.求出某区间每一个数的和 输入输出格式 输入格 ...
随机推荐
- 2019年猪年海报PSD模板-第六部分
14套精美猪年海报,免费猪年海报,下载地址:百度网盘,https://pan.baidu.com/s/1WdlIiIdj1VVWxI4je0ebKw
- 三分钟小课堂-----------------docker(三)增删改查命令
主要为docker容器的增删改查命令 1 创建容器: docker run -it --name 别名 image_name /bin/bash --name 别名 -d 后台 -t ...
- 加油吧 骚年QAQ
本随笔文章,由个人博客(鸟不拉屎)转移至博客园 写于:2017 年 11 月 08 日 原地址:https://niaobulashi.com/archives/fighting.html --- 想 ...
- python numpy数据相减
numpy数据相减,a和b两者shape要一样,然后是对应的位置相减.要不然,a的shape可以是(1,m),注意m要等于b的列数. import numpy as np a = [ [0, 1, 2 ...
- 在mesh client示例中加入spi_slave接口(without IDE)
在mesh client示例中加入spi_slave接口(without IDE) 主要是理解cmake构建的过程,然后修改工程中的inlcude路径及c源文件. 1. 解压mesh_sdk unzi ...
- C#调用mingw的so库时无法加载DLL###.so 找不到指定的模块
使用C#调用mingw的so,报了c# 无法加载DLL“###.so”,: 找不到指定的程序. (异常来自 HRESULT:0x8007007E)开始以为是dll路径问题,使用全路径确认正确后仍然无法 ...
- CentOS Openvpn搭建以及 linux&&windows客户端的连接
本文参考:http://www.centoscn.com/CentosServer/test/2014/1120/4153.html 一. Server安装准备 (CentOS release ...
- 自测之Lesson5:标准I/O
题目:使用perror函数和strerror函数编写一个程序. 程序代码: #include <stdio.h> #include <errno.h> #include < ...
- “Hello world!”团队—文案+美工
★★★本次采访我们随机选取5位不同的潜在用户,随机选取地点进行了本次采访. (一)项目有关内容: 大家好,我们是Hello World团队.我们组目前正在开发一个飞机大战的小游戏大家应该在小时候都玩过 ...
- 新人学PHP,认为手动搭建环境而苦恼吗?这篇文章告诉你多简单!
本教程适用于初学PHP,想了解手动搭建PHP环境的童鞋. 一键环境和高手勿喷. 本教程以下列版本软件为例: 所需软件目录 我在这里的目录结构是(个人习惯) 安装与配置 apache 双击安装Apach ...