Luogu4191:[CTSC2010]性能优化

传送门
题目翻译:给定两个 n n n 次多项式 A , B A,B A,B 和一个整数 C C C,求 A × B C A\times B^C A×BC 在模 x n x^n xn 意义下的卷积
显然就是个循环卷积,所以只要代入 ω n k \omega_n^{k} ωnk 进去求出点值,然后插值就好了
??? n n n 不是 2 k 2^k 2k 的形式,不能直接 N T T NTT NTT
怎么办呢?
根据题目性质,可以把 n n n 拆成 2 a 1 3 a 2 5 a 3 7 a 4 2^{a_1}3^{a_2}5^{a_3}7^{a_4} 2a13a25a37a4 的形式
这启示我们每次不是每次分成两半而是拆分成 3 / 5 / 7 3/5/7 3/5/7 次,然后再合并点值
F ( x ) = ∑ a i x i , F r ( x ) = ∑ a i p + r x i F(x)=\sum a_ix^i,F_r(x)=\sum a_{ip+r}x^i F(x)=aixiFr(x)=aip+rxi
那么 F ( x ) = ∑ x r F ( x p ) F(x)=\sum x^rF(x^p) F(x)=xrF(xp)
根据单位复数的性质(消去引理和折半引理)那么
F ( ω n a n + b ) = ∑ ω n p ( a n + b ) r F r ( w n b ) F(\omega_n^{an+b})=\sum \omega_{np}^{(an+b)r}F_r(w_n^b) F(ωnan+b)=ωnp(an+b)rFr(wnb)
那么只需要写一个每次分 p p p 份的 F F T FFT FFT 就好了

# include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn(5e5 + 5);

int n, c, a[maxn], b[maxn], tmp[maxn], g, pri[233333], tot, pw[2][maxn], mod, r[maxn];

inline int Pow(ll x, int y) {
	register ll ret = 1;
	for (; y; y >>= 1, x = x * x % mod)
		if (y & 1) ret = ret * x % mod;
	return ret;
}

inline void Inc(int &x, int y) {
	x = x + y >= mod ? x + y - mod : x + y;
}

int Dfs(int s, int p, int cur, int blk) {
	if (cur == tot + 1) return s + p;
	register int nxt;
	nxt = blk / pri[cur];
	return Dfs(s + nxt * (p % pri[cur]), (p - p % pri[cur]) / pri[cur], cur + 1, nxt);
}

inline void DFT(int *p, int opt) {
	register int i, j, k, l, q, t, cur;
	for (i = 0; i < n; ++i) tmp[r[i]] = p[i];
	for (i = 0; i < n; ++i) p[i] = tmp[i], tmp[i] = 0;
	for (i = 1, cur = tot; i < n; i *= pri[cur], --cur) {
		for (t = i * pri[cur], j = 0; j < n; j += t)
			for (k = 0; k < t; k += i)
				for (l = 0; l < i; ++l)
					for (q = 0; q < pri[cur]; ++q)
						Inc(tmp[j + k + l], (ll)pw[opt == -1][n / t * (k + l) * q % n] * p[j + i * q + l] % mod);
		for (j = 0; j < n; ++j) p[j] = tmp[j], tmp[j] = 0;
	}
	if (opt == -1) for (c = Pow(n, mod - 2), i = 0; i < n; ++i) p[i] = (ll)p[i] * c % mod;
}

int main() {
	register int i, j, x;
	scanf("%d%d", &n, &c), mod = n + 1;
	for (x = n, i = 2; i * i <= x; ++i)
		while (x % i == 0) pri[++tot] = i, x /= i;
	if (x > 1) pri[++tot] = x;
	for (i = 2; ; ++i) {
		for (g = i, j = 1; g && j <= tot; ++j)
			if (Pow(g, n / pri[j]) == 1) g = 0;
		if (g) break;
	}
	for (i = 0; i < n; ++i) scanf("%d", &a[i]);
	for (i = 0; i < n; ++i) scanf("%d", &b[i]);
	pw[0][0] = pw[1][0] = 1, pw[0][1] = g, pw[1][1] = Pow(g, mod - 2);
	for (i = 2; i < n; ++i) pw[0][i] = (ll)pw[0][i - 1] * g % mod, pw[1][i] = (ll)pw[1][i - 1] * pw[1][1] % mod;
	for (i = 0; i < n; ++i) r[i] = Dfs(0, i, 1, n);
	DFT(a, 1), DFT(b, 1);
	for (i = 0; i < n; ++i) a[i] = (ll)a[i] * Pow(b[i], c) % mod;
	DFT(a, -1);
	for (i = 0; i < n; ++i) printf("%d\n", a[i]);
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值