P5277 【模板】多项式开根(加强版)

题目背景

模板题,无背景

题目描述

给定一个 $n-1$ 次多项式 $A(x)$ ,求一个在 $\bmod\ {x^n}$ 意义下的多项式 $B(x)$ ,使得 $B^2(x)\equiv A(x)\pmod {x^n}$。 多项式的系数在 $\bmod\ 998244353$ 的意义下进行运算。

输入格式

第一行一个正整数 $n$。 接下来 $n$ 个整数,依次表示多项式的系数 $a_0,a_1,\ldots ,a_{n-1}$。 **不保证 $a_0=1$,但保证 $a_0$ 是 $\bmod\ 998244353$ 下的二次剩余。**

输出格式

输出 $n$ 个非负整数,表示答案多项式的系数 $b_0,b_1,\ldots ,b_{n-1}$。如有多解,输出**系数序列**(而非字符序列)字典序最小的。

说明/提示

对于 $25\%$ 的数据,有 $n \leq 1000$。 对于 $50\%$ 的数据,有 $n \leq 10^4$。 对于 $75\%$ 的数据,有 $n \leq 5\times 10^4$。 对于 $100\%$ 的数据,有 $n \leq 10^5,a_i \in [0,998244352] \cap \mathbb{Z}$。