FFT模板

const int maxn = 2e5 + 10;
const double pi = acos(-1);

ll c[maxn];
ll d[maxn];

struct cplx {
    double r, i;
    cplx(double r = 0, double i = 0) : r(r), i(i) {}
    cplx operator + (const cplx &a) const {
        return cplx(r + a.r, i + a.i);
    }
    cplx operator - (const cplx &a) const {
        return cplx(r - a.r, i - a.i);
    }
    cplx operator * (const cplx &a) const {
        return cplx(r * a.r - i * a.i, r * a.i + i * a.r);
    }
} a[maxn << 2], b[maxn << 2];

void rader(cplx *f, int len)
{
    int j = len >> 1;
    for(int i = 1; i < len - 1; i++)
    {
        if(i < j) swap(f[i], f[j]);
        int k = len >> 1;
        while(j >= k)
        {
            j -= k;
            k >>= 1;
        }
        if(j < k) j += k;
    }
}

void FFT(cplx *f, int len, int t)
{
    rader(f, len);
    for(int h = 2; h <= len; h <<= 1)
    {
        cplx wn(cos(- t * 2 * pi / h), sin(- t * 2 * pi / h));
        for(int j = 0; j < len; j += h)
        {
            cplx e(1, 0);
            for(int k = j; k < j + h / 2; k++)
            {
                cplx u = f[k];
                cplx v = e * f[k + h / 2];
                f[k] = u + v;
                f[k + h / 2] = u - v;
                e = e * wn;
            }
        }
    }
    if(t == -1) for(int i = 0; i < len; i++) f[i].r /= len;
}

void Conv(cplx *a, cplx *b, int len)
{
    FFT(a, len, 1);
    FFT(b, len, 1);
    for(int i = 0; i < len; i++) a[i] = a[i] * b[i];
    FFT(a, len, -1);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值