FFT模板

最新推荐文章于 2022-01-09 17:46:45 发布
最新推荐文章于 2022-01-09 17:46:45 发布
阅读量66 收藏
点赞数
原文链接：http://www.cnblogs.com/LiHior/p/9505036.html
版权

FFT 实际是 DFT 的一种快速实现方法

可以将多项式的乘法从 O(n^2) 优化到 O(nlogn)

暂时没有看到很好的科普文章、原理自行百度吧

 

#define L(x) (1 << (x))
const double PI = acos(-1.0);
const int maxn = (1<<17) + (int)1e3;
double ax[maxn], ay[maxn], bx[maxn], by[maxn];
int revv(int x, int bits)
{
    int ret = 0;
    for (int i = 0; i < bits; i++){
        ret <<= 1;
        ret |= x & 1;
        x >>= 1;
    }
    return ret;
}

void fft(double * a, double * b, int n, bool rev)
{
    int bits = 0;
    while (1 << bits < n) ++bits;
    for (int i = 0; i < n; i++){
        int j = revv(i, bits);
        if (i < j) swap(a[i], a[j]), swap(b[i], b[j]);
    }

    for (int len = 2; len <= n; len <<= 1){
        int half = len >> 1;
        double wmx = cos(2 * PI / len), wmy = sin(2 * PI / len);
        if (rev) wmy = -wmy;
        for (int i = 0; i < n; i += len){
            double wx = 1, wy = 0;
            for (int j = 0; j < half; j++){
                double cx = a[i + j], cy = b[i + j];
                double dx = a[i + j + half], dy = b[i + j + half];
                double ex = dx * wx - dy * wy, ey = dx * wy + dy * wx;
                a[i + j] = cx + ex, b[i + j] = cy + ey;
                a[i + j + half] = cx - ex, b[i + j + half] = cy - ey;
                double wnx = wx * wmx - wy * wmy, wny = wx * wmy + wy * wmx;
                wx = wnx, wy = wny;
            }
        }
    }
    if (rev){
        for (int i = 0; i < n; i++)
            a[i] /= n, b[i] /= n;
    }
}

int Convolution(int a[],int na,int b[],int nb,int ans[]) //两个数组求卷积,有时ans数组要开成long long
{
    int len = max(na, nb), ln;
    for(ln=0; L(ln)<len; ++ln);
    len=L(++ln);
    for (int i = 0; i < len ; ++i){
        if (i >= na) ax[i] = 0, ay[i] =0;
        else ax[i] = a[i], ay[i] = 0;
    }
    fft(ax, ay, len, 0);
    for (int i = 0; i < len; ++i){
        if (i >= nb) bx[i] = 0, by[i] = 0;
        else bx[i] = b[i], by[i] = 0;
    }
    fft(bx, by, len, 0);
    for (int i = 0; i < len; ++i){
        double cx = ax[i] * bx[i] - ay[i] * by[i];
        double cy = ax[i] * by[i] + ay[i] * bx[i];
        ax[i] = cx, ay[i] = cy;
    }
    fft(ax, ay, len, 1);
    for (int i = 0; i < len; ++i)
        ans[i] = (int)(ax[i] + 0.5);
    return len;
}


int Convolution_self(long long a[], int na, int ans[]) //自己跟自己求卷积,有时候ans数组要开成long long
{
    int len = na, ln;
    for(ln = 0; L(ln) < na; ++ln);
    len=L(++ln);
    for(int i = 0; i < len; ++i){
        if (i >= na) ax[i] = 0, ay[i] = 0;
        else ax[i] = a[i], ay[i] = 0;
    }
    fft(ax, ay, len, 0);
    for(int i=0; i<len; ++i){
        double cx = ax[i] * ax[i] - ay[i] * ay[i];
        double cy = 2 * ax[i] * ay[i];
        ax[i] = cx, ay[i] = cy;
    }
    fft(ax, ay, len, 1);

    for(int i=0; i<len; ++i)
        ans[i] = ax[i] + 0.5;
    return len;
}
View Code

 

模板题

 

#include<bits/stdc++.h>
#define LL long long
#define ULL unsigned long long

#define scl(i) scanf("%lld", &i)
#define scll(i, j) scanf("%lld %lld", &i, &j)
#define sclll(i, j, k) scanf("%lld %lld %lld", &i, &j, &k)
#define scllll(i, j, k, l) scanf("%lld %lld %lld %lld", &i, &j, &k, &l)

#define scs(i) scanf("%s", i)
#define sci(i) scanf("%d", &i)
#define scd(i) scanf("%lf", &i)
#define scIl(i) scanf("%I64d", &i)
#define scii(i, j) scanf("%d %d", &i, &j)
#define scdd(i, j) scanf("%lf %lf", &i, &j)
#define scIll(i, j) scanf("%I64d %I64d", &i, &j)
#define sciii(i, j, k) scanf("%d %d %d", &i, &j, &k)
#define scddd(i, j, k) scanf("%lf %lf %lf", &i, &j, &k)
#define scIlll(i, j, k) scanf("%I64d %I64d %I64d", &i, &j, &k)
#define sciiii(i, j, k, l) scanf("%d %d %d %d", &i, &j, &k, &l)
#define scdddd(i, j, k, l) scanf("%lf %lf %lf %lf", &i, &j, &k, &l)
#define scIllll(i, j, k, l) scanf("%I64d %I64d %I64d %I64d", &i, &j, &k, &l)

#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
#define lowbit(i) (i & (-i))
#define mem(i, j) memset(i, j, sizeof(i))

#define fir first
#define sec second
#define VI vector<int>
#define ins(i) insert(i)
#define pb(i) push_back(i)
#define pii pair<int, int>
#define VL vector<long long>
#define mk(i, j) make_pair(i, j)
#define all(i) i.begin(), i.end()
#define pll pair<long long, long long>

#define _TIME 0
#define _INPUT 0
#define _OUTPUT 0
clock_t START, END;
void __stTIME();
void __enTIME();
void __IOPUT();
using namespace std;

#define L(x) (1 << (x))
const double PI = acos(-1.0);
const int maxn = (1<<20) + (int)1e3;
double ax[maxn], ay[maxn], bx[maxn], by[maxn];
int revv(int x, int bits)
{
    int ret = 0;
    for (int i = 0; i < bits; i++){
        ret <<= 1;
        ret |= x & 1;
        x >>= 1;
    }
    return ret;
}

void fft(double * a, double * b, int n, bool rev)
{
    int bits = 0;
    while (1 << bits < n) ++bits;
    for (int i = 0; i < n; i++){
        int j = revv(i, bits);
        if (i < j) swap(a[i], a[j]), swap(b[i], b[j]);
    }

    for (int len = 2; len <= n; len <<= 1){
        int half = len >> 1;
        double wmx = cos(2 * PI / len), wmy = sin(2 * PI / len);
        if (rev) wmy = -wmy;
        for (int i = 0; i < n; i += len){
            double wx = 1, wy = 0;
            for (int j = 0; j < half; j++){
                double cx = a[i + j], cy = b[i + j];
                double dx = a[i + j + half], dy = b[i + j + half];
                double ex = dx * wx - dy * wy, ey = dx * wy + dy * wx;
                a[i + j] = cx + ex, b[i + j] = cy + ey;
                a[i + j + half] = cx - ex, b[i + j + half] = cy - ey;
                double wnx = wx * wmx - wy * wmy, wny = wx * wmy + wy * wmx;
                wx = wnx, wy = wny;
            }
        }
    }
    if (rev){
        for (int i = 0; i < n; i++)
            a[i] /= n, b[i] /= n;
    }
}

int Convolution(int a[],int na,int b[],int nb,int ans[]) //两个数组求卷积,有时ans数组要开成long long
{
    int len = max(na, nb), ln;
    for(ln=0; L(ln)<len; ++ln);
    len=L(++ln);
    for (int i = 0; i < len ; ++i){
        if (i >= na) ax[i] = 0, ay[i] =0;
        else ax[i] = a[i], ay[i] = 0;
    }
    fft(ax, ay, len, 0);
    for (int i = 0; i < len; ++i){
        if (i >= nb) bx[i] = 0, by[i] = 0;
        else bx[i] = b[i], by[i] = 0;
    }
    fft(bx, by, len, 0);
    for (int i = 0; i < len; ++i){
        double cx = ax[i] * bx[i] - ay[i] * by[i];
        double cy = ax[i] * by[i] + ay[i] * bx[i];
        ax[i] = cx, ay[i] = cy;
    }
    fft(ax, ay, len, 1);
    for (int i = 0; i < len; ++i)
        ans[i] = (int)(ax[i] + 0.5);
    return len;
}


int Convolution_self(long long a[], int na, int ans[]) //自己跟自己求卷积,有时候ans数组要开成long long
{
    int len = na, ln;
    for(ln = 0; L(ln) < na; ++ln);
    len=L(++ln);
    for(int i = 0; i < len; ++i){
        if (i >= na) ax[i] = 0, ay[i] = 0;
        else ax[i] = a[i], ay[i] = 0;
    }
    fft(ax, ay, len, 0);
    for(int i=0; i<len; ++i){
        double cx = ax[i] * ax[i] - ay[i] * ay[i];
        double cy = 2 * ax[i] * ay[i];
        ax[i] = cx, ay[i] = cy;
    }
    fft(ax, ay, len, 1);

    for(int i=0; i<len; ++i)
        ans[i] = ax[i] + 0.5;
    return len;
}

int a[maxn], b[maxn], c[maxn];
int main(void){__stTIME();__IOPUT();

    int n, m;
    scii(n, m);

    for(int i=0; i<=n; i++) sci(a[i]);
    for(int i=0; i<=m; i++) sci(b[i]);

    int len = Convolution(a, n+1, b, m+1, c);

    for(int i=0; i<n+m+1; i++) printf("%d ", c[i]); puts("");


__enTIME();return 0;}


void __stTIME()
{
    #if _TIME
        START = clock();
    #endif
}

void __enTIME()
{
    #if _TIME
        END = clock();
        cerr<<"execute time = "<<(double)(END-START)/CLOCKS_PER_SEC<<endl;
    #endif
}

void __IOPUT()
{
    #if _INPUT
        freopen("in.txt", "r", stdin);
    #endif
    #if _OUTPUT
        freopen("out.txt", "w", stdout);
    #endif
}
View Code

 

转载于:https://www.cnblogs.com/LiHior/p/9505036.html

ajumn60282
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
快速傅里叶变化（FFT）含模板
zhz的博客
07-04 498
快速傅里叶变化（FFT） 资料来源：https://www.bilibili.com/video/BV1Y7411W73U 前置知识索引 FFT的核心算法是基于复数和单位根的，并且在算法竞赛中FFT的最普遍应用是求i多项式乘法，或者是卷积。 复数 Complex Number ​ 定义：z=a+bi其中a,b∈R，i=n加法法则：(a+bi)+(c+di)=(a+c)+(b+d)i乘法法则：（a+bi)(c+di)=(ac−bd)+(ad+bc)i欧拉定理：eiθ=cos⁡θ+isin⁡θ 定义：z =
【数论】fft模板
qq_23096319的博客
03-04 475
在O(nlogn)时间内将系数表示多项式转化为点值表示，n个点确定一条n−1次的多项式曲线，后者可以O(n)时间内完成高精度乘法。而这n个点的取法是有讲究的，，对，将x的取值扩展到复数域，这些点也称为单位根，定义为ωnk​cosnk​2πisinnk​2π，默认n为2的整数次幂。对于多项式Axa0​a1​xa2​x2...an−1​xn−1将项根据下标的奇偶性分别为两部分，Axa0​a2​。
FFT模板-附详细注释
pursuit的博客
08-05 835
学习自：快速傅里叶变换(FFT)详解,【学习笔记】超简单的快速傅里叶变换（FFT）（含全套证明）。 FFT模板 /** *@filename:FFT *@author: pursuit *@created: 2021-08-05 20:57 **/ #include <bits/stdc++.h> #define debug(a) cout << "debug : " << (#a)<< " = " << a << en
FFT（快速傅里叶变换）模板
Cosmic_Tree的博客
07-23 361
学习链接1：点击进入 学习链接2：点击进入 主要用处 O(nlogn)计算多项式乘法。 DFT（离散傅里叶变换）：多项式转化成“离散的点”，系数表示法变成点值表示法. IDFT（离散傅里叶反变换）把“离散的点”还原回多项式，点值表示法变成系数表示法。 … … 代码（模板） #include<iostream> #include<string> #include<map> #include<set> //#include<unordered_map>
FFT滤波实验
03-27
此外，还可以自定义滤波器模板，以满足特定的滤波需求。 4. **逆傅里叶变换**：应用逆FFT (IFFT) 将处理后的频域图像转换回时域，生成滤波后的图像。 5. **后处理**：最后，可能需要对结果图像进行一些后处理，...
fft.rar_fft_fft变换
09-20
**快速傅里叶变换(FFT)详解** 快速傅里叶变换（Fast Fourier Transform，简称FFT）是一种用于计算离散傅里叶变换（Discrete ...在实际编程中，结合`COMPLEX`模板和`VECTOR`等数据结构，可以方便地实现和应用FFT算法。
fft.rar_fft_matlab fft
09-22
使用模板类和STL容器可以帮助提高代码的可读性和复用性。 5. 调试和优化C++ FFT：为了提升C++ FFT的效率，开发者可能需要进行性能分析，识别瓶颈，并使用诸如OpenMP这样的库进行并行化，或者采用其他优化技术，如...
FFT
03-29
在C++中，可以使用标准模板库(STL)以及复数类来实现FFT。以下是一个简单的基于Cooley-Tukey算法的FFT实现步骤： 1. 定义复数类：包含实部和虚部，提供复数运算的重载方法。 2. 编写递归的FFT函数：接受一个复数...
洛谷 P3803 FFT
Ambrumf的博客
01-09 467
思路：FFT模板 #include <bits/stdc++.h> using namespace std; typedef long long ll; const double PI = acos(-1); const int N = 1e7 + 7; int n, m; int res, ans[N]; int limit = 1;// 补齐的2的整数幂N int L;// 二进制的位数 int R[N];// 二进制翻转 struct Complex { double x, y;
FFT快速傅里叶变换-递归版-带注释模板
03-05 250
题链：uoj#34多项式乘法 题意：给你两个多项式，请输出乘起来后的多项式。 //打法照抄hyc并复制了部分注释 真·存模板系列 #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<iostream&...
hdu 1402 （FFT 模版）
tjf的专栏
08-26 718
/* algorithm : High-Precision FFT Memory 7092K Time 359MS Language G++ code by : zhyu */ #include #include #include #include #define N 200005 #define pi acos(-1.0) // PI值 using namespace std
FFT（最详细最通俗的入门手册）
热门推荐
逐梦者
07-24 6万+
声明首先，我需要声明，本文是在转载的基础上稍微修饰的，经过原创作者 ZLH_HHHH（佐理慧学姐） 的许可方才转载并修饰的，由于我就是初学者，并且是数学渣滓，所以我学姐建议我写一下残疾人手册，我当然是欣然接受！！！正文：文章写的有点急。有错误的地方望指出 我学习 FFT 是一个比较慢的过程。 期间反反复复。 我写这篇博文只是一个非常浅显的理解。同时也可以帮助初学者在学习FFT的时候。有所偏重。避免
模板 FFT 快速傅里叶变换
guapisolo的博客
09-18 181
FFT模板，原理不难，优质讲解很多，但证明很难看太不懂 这模板题在bzoj竟然是土豪题，服了 #include &lt;cmath&gt; #include &lt;cstdio&gt; #include &lt;cstring&gt; #include &lt;algorithm&gt; #define dd double #define ll long long #define N (...
fft模板
slow_yangyang的博客
09-29 3177
整理了一下目前所写过的fft题目（使用kuangbin模板） 首先需要明白一个概念，fft是用来求卷积的，他可以把离散的xx变成连续的xx，然后把n^2的时间复杂度变成nlogn。 比如现在计算28*76的值就可以使用卷积计算。 第一步：先把76逆置成67 第二步： 28   67  得8*6=48，放入a数组，a[0]=48 第三步： 28 67  得2*6+8*7=6
【FFT快速傅里叶变换】【模板】
herongwei 的 BLOG
04-29 3160
【题目链接】：http://uoj.ac/problem/34 【题目描述】： #34. 多项式乘法  统计  描述 提交 自定义测试 这是一道模板题。 给你两个多项式，请输出乘起来后的多项式。 输入格式 第一行两个整数 nn 和 mm，分别表示两个多项式的次数。 第二行 n+1n+1 个整数，分别表示第一个多项式的 00 到 nn 次项前的系数。 第三
请用C++写一个FFT模板
最新发布
04-11
好的，以下是C语言中的FFT模板： ```c #include #include typedef struct _complex { double real, imag; } complex; const double PI = atan(1.0) * 4; void fft(complex *x, int N) { if (N ) return; ...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值