HDU 6061 RXD and functions（NTT）

RXD and functions

首先是有一个结论，对多项式做任意多次 transformation ，其结果跟做一次 transformation Tr(f,∑i=1mai)Tr(f, \sum\limits_{i = 1} ^{m} a_i)Tr(f,i=1∑mai)的结果是一样的，所以我们约定a=−∑i=1maia = -\sum\limits_{i = 1} ^{m} a_ia=−i=1∑mai。
g=f(x+a)∑i=0nci(x+a)i∑i=0n∑j=0icii!j!(i−j)!xjai−j我们考虑只求这个多项式的xj系数gj=∑i=jncii!j!(i−j)!ai−jgj=∑i=0n−jci+j(i+j)!j!i!aigj=1j!∑i=0n−j(ci+j(i+j)!)×(aii!)设f(n)=cnn!,h(n)=ann!不难发现后项像是多项式卷积的形式，考虑翻转aii，即i=n−igj=1j!∑i=0n−jf(i+j)h(n−i)这就是一个完美的卷积形式了，于是可以NTT求解g = f(x + a)\\ \sum_{i = 0} ^{n} c_i(x + a) ^ i\\ \sum_{i = 0} ^{n} \sum_{j = 0} ^{i}c_i \frac{i!}{j!(i - j)!} x ^ j a ^ {i - j}\\ 我们考虑只求这个多项式的x ^ j系数\\ g_j = \sum_{i = j} ^{n} c_i \frac{i!}{j!(i - j)!}a ^{i - j}\\ g_j = \sum_{i = 0} ^{n - j} c_{i + j} \frac{(i + j)!}{j! i!} a ^ i\\ g_j = \frac{1}{j!} \sum_{i = 0} ^{n - j} (c_{i + j} (i + j)!) \times (\frac{a ^i}{i!})\\ 设f(n) = c_{n} n!, h(n) = \frac{a ^ n}{n!} \\ 不难发现后项像是多项式卷积的形式，考虑翻转\frac{a ^ i}{ i}，即i = n - i\\ g_j = \frac{1}{j!} \sum_{i = 0} ^{n - j} f(i + j) h(n - i)\\ 这就是一个完美的卷积形式了，于是可以NTT求解 g=f(x+a)i=0∑nci(x+a)ii=0∑nj=0∑icij!(i−j)!i!xjai−j我们考虑只求这个多项式的xj系数gj=i=j∑ncij!(i−j)!i!ai−jgj=i=0∑n−jci+jj!i!(i+j)!aigj=j!1i=0∑n−j(ci+j(i+j)!)×(i!ai)设f(n)=cnn!,h(n)=n!an不难发现后项像是多项式卷积的形式，考虑翻转iai，即i=n−igj=j!1i=0∑n−jf(i+j)h(n−i)这就是一个完美的卷积形式了，于是可以NTT求解

#include <bits/stdc++.h>using namespace std;const int N = 1e6 + 10, mod = 998244353;int r[N];int quick_pow(int a, int n) {int ans = 1;while (n) {if (n & 1) {ans = 1ll * a * ans % mod;}a = 1ll * a * a % mod;n >>= 1;}return ans;
}void get_r(int lim) {for (int i = 0; i < lim; i++) {r[i] = (i & 1) * (lim >> 1) + (r[i >> 1] >> 1);}
}void NTT(int *f, int lim, int rev) {for (int i = 0; i < lim; i++) {if (i < r[i]) {swap(f[i], f[r[i]]);}}for (int mid = 1; mid < lim; mid <<= 1) {int wn = quick_pow(3, (mod - 1) / (mid << 1));for (int len = mid << 1, cur = 0; cur < lim; cur += len) {int w = 1;for (int k = 0; k < mid; k++, w = 1ll * w * wn % mod) {int x = f[cur + k], y = 1ll * w * f[cur + mid + k] % mod;f[cur + k] = (x + y) % mod, f[cur + mid + k] = (x - y + mod) % mod;}}}if (rev == -1) {int inv = quick_pow(lim, mod - 2);reverse(f + 1, f + lim);for (int i = 0; i < lim; i++) {f[i] = 1ll * f[i] * inv % mod;}}
}int a[N], c[N], A[N], B[N], fac[N], ifac[N], n, m, sum;void init() {fac[0] = 1;for (int i = 1; i < N; i++) {fac[i] = 1ll * fac[i - 1] * i % mod;}ifac[N - 1] = quick_pow(fac[N - 1], mod - 2);for (int i = N - 2; i >= 0; i--) {ifac[i] = 1ll * ifac[i + 1] * (i + 1) % mod;}
}int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);init();while (scanf("%d", &n) != EOF) {for (int i = 0; i <= n; i++) {scanf("%d", &c[i]);}scanf("%d", &m);sum = 0;for (int i = 1; i <= m; i++) {scanf("%d", &a[i]);sum = (sum + a[i]) % mod;}sum = (mod - sum) % mod;int cur = 1;for (int i = 0; i <= n; i++) {A[i] = 1ll * c[i] * fac[i] % mod;B[n - i] = 1ll * ifac[i] * cur % mod;cur = 1ll * cur * sum % mod;}int lim = 1;while (lim <= 2 * n) {lim <<= 1;}get_r(lim);NTT(A, lim, 1);NTT(B, lim, 1);for (int i = 0; i < lim; i++) {A[i] = 1ll * A[i] * B[i] % mod;}NTT(A, lim, -1);for (int i = 0; i <= n; i++) {printf("%lld ", 1ll * ifac[i] * A[n + i] % mod);}puts("");for (int i = 0; i < lim; i++) {A[i] = B[i] = 0;}}return 0;
}

HDU 6061 RXD and functions（NTT）相关推荐

【推导】【NTT】hdu6061 RXD and functions（NTT）
题意:给定一个n次多项式f(x)的各项系数,让你求f(x-Σai)的各项系数. http://blog.csdn.net/v5zsq/article/details/76780053 推导才是最关键的 ...
2017 多校3 hdu 6061 RXD and functions
2017 多校3 hdu 6061 RXD and functions(FFT) 题意: 给一个函数\(f(x)=\sum_{i=0}^{n}c_i \cdot x^{i}\) 求\(g(x) = f ...
【学习笔记】超简单的快速数论变换（NTT）（FFT的优化）（含全套证明）
整理的算法模板合集: ACM模板目录一.前置知识二.快速数论变换(NTT) 三.NTT证明(和FFT的关系) 四.NTT模板数组形式的实现 vector形式的实现点我看多项式全家桶(●^◡_ ...
HDU各种比赛题题解（一）
HDU各种比赛题题解(一) Gardon-DYGG Contest 1 HDU1178 Heritage from father[水题] - 海岛Blog - CSDN博客 HDU1181 变形课[D ...
【HDU 2612 Find a Way（BFS）】(兼BFS入门笔记)
[HDU 2612 Find a Way(BFS)](兼BFS入门笔记) 原题入口: http://acm.hdu.edu.cn/showproblem.php?pid=2612 第一篇在CSDN的博 ...
【luogu P3803】【模板】多项式乘法（NTT）
[模板]多项式乘法(NTT) 题目链接:luogu P3803 题目大意给你两个多项式,要你求它们的卷积. 思路这次我们写 NTT 的做法. 它的优点就是它可以取模,而且不会有精度问题,而且会比 ...
【HDU 3400】Line belt（三分法）
题目链接题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=3400 题意有两条传送带AB和CD,移动速度分别为p,q. 除了传送带的其他区域移动速度为r ...
快速数论变换（NTT）
刚学完FFT,干脆把NTT也学了算了 (一)预备知识关于原根,这里说得蛮详细的百度百科为什么使用原根呢?为什么原根可以替代\(\omega_{n}\)呢?想知道为什么就看here NTT用到的各种 ...
HDU 1222 Wolf and Rabbit（gcd）
HDU 1222 Wolf and Rabbit (最大公约数)解题报告题目链接:http://acm.hust.edu.cn/vjudge/contest/view.action?cid= ...

HDU 6061 RXD and functions（NTT）

RXD and functions

HDU 6061 RXD and functions（NTT）相关推荐

最新文章

热门文章