ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

FFT_应用和例题

2019-12-12 10:00:29  阅读:315  来源: 互联网

标签:aligned frac 卷积 sum FFT 应用 otimes 例题


卷积

现有两个定义在 N 上的函数 \(f(n),g(n)\),定义 \(f\) 和 \(g\) 的卷积(convolution)为 \(f \otimes g\)
\[ (f \otimes g)(n) = \sum_{i=0}^n f(i)g(n-i) \]

示意图: from http://blog.miskcoo.com/2015/04/polynomial-multiplication-and-fast-fourier-transform#i-17

示意图

考虑两多项式 \(A, B\) 的乘积 \(C\), \(c(x) = \sum_{i=0}^{x} a(i) \cdot b(x - i)\)
系数记为卷积形式

于是计算卷积 \((f \otimes g)(n)\) 就可以把 \(f, g\) 的值直接作为系数写成两个多项式, 然后 FFT 计算多项式乘积, 得到的系数的前 \(n\) 项即为所求

BZOJ3527[ZJOI2014]力

题意:
给出 \(n\) 个数 \(q_i\) ,给出 \(F_j\) 的定义如下:
\[ F_j = \sum_{i < j}\frac{q_i q_j}{(i - j) ^ 2} - \sum_{i > j}\frac{q_i q_j}{(i - j) ^ 2} \]
令 \(E_j = F_j / q_j\) , 求 \(E_j\)

Sol:

因为知道是卷积的例题了, 所以想着把这个式子往卷积的方向靠

\[ \begin{aligned} E_j = \sum_{i < j}\frac{q_i}{(i - j) ^ 2} - \sum_{i > j}\frac{q_i}{(i - j) ^ 2} \end{aligned} \]

考虑分母当做系数, 再考虑下标和为 \(j\) , 写成这样

\[ \begin{aligned} E_j = \sum_{i < j}\frac{1}{(j - i) ^ 2}q_i - \sum_{i > j}\frac{1}{(j - i) ^ 2}q_i \end{aligned} \]

一开始想把两个一起做, 发现写不出两个函数, 于是考虑分开做
显然 \(\sum_{i < j}\frac{1}{(j - i) ^ 2}q_i\) 就是 \(f(n) = q_n\) 和 \(g(n) = \frac{1}{n^2}\) 的卷积
然后后面一项同理, 把 \(f(n)\) 翻转一下即可

然后跑 FFT

double ans[MAXN], q[MAXN];

/*
20191212
0859~0922~0939
BZOJ3527 FFT
 */

int main()
{
    scanf("%d", &lena);
    for (int i = 0; i < lena; ++ i) 
    {
    scanf("%lf", &a[i].x); q[lena - 1 - i] = a[i].x;
    b[i].x = (i == 0 ? 0.0 : 1.0 / i / i);
    }
    while ((1 << dgt) < lena * 2) ++ dgt;
    n = 1 << dgt;
    init(n, dgt);
    FFT(b, n ,1);
    FFT(a, n, 1);
    for (int i = 0; i < n; ++ i) a[i] = a[i] * b[i];
    FFT(a, n, -1);
    for (int i = 0; i < lena; ++ i) ans[i] += a[i].x / n;
    for (int i = 0; i < n; ++ i) a[i].x = q[i], a[i].y = 0;
    FFT(a, n, 1);
    for (int i = 0; i < n; ++ i) a[i] = a[i] * b[i];
    FFT(a, n, -1);
    for (int i = 0; i < lena; ++ i) ans[i] -= a[lena - 1 - i].x / n;
    for (int i = 0; i < lena; ++ i) printf("%.3f\n", ans[i]);
    return 0;
}
/*
3
1 2 3
 */

标签:aligned,frac,卷积,sum,FFT,应用,otimes,例题
来源: https://www.cnblogs.com/Kuonji/p/12027519.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有