分治FFT/NTT

粘板子:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MOD = 998244353;
const int N = 100050;
const int M = N*3;
template<typename T>
inline void read(T&x)
{
    T f = 1,c = 0;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){c=c*10+ch-‘0‘;ch=getchar();}
    x = f*c;
}
template<typename T>inline void Mod(T&x){if(x>=MOD)x-=MOD;}
int fastpow(int x,int y)
{
    int ret = 1;
    while(y)
    {
        if(y&1)ret=1ll*ret*x%MOD;
        x=1ll*x*x%MOD;y>>=1;
    }
    return ret;
}
int inv(int x){return fastpow(x,MOD-2);}
int to[M],lim,L,LL[M];
void init(int len)
{
    lim=LL[2]=1;
    while(lim<len)lim<<=1,LL[lim<<1]=LL[lim]+1;
}
void get_lim(int len)
{
    lim = len,L = LL[len];
    for(int i=1;i<=lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));
}
void ntt(int*a,int len,int k)
{
    for(int i=0;i<len;i++)
        if(i<to[i])swap(a[i],a[to[i]]);
    for(int i=1;i<len;i<<=1)
    {
        int w0 = fastpow(3,(MOD-1)/(i<<1));
        for(int j=0;j<len;j+=(i<<1))
        {
            int w = 1;
            for(int o=0;o<i;o++,w=1ll*w*w0%MOD)
            {
                int w1 = a[j+o],w2 = 1ll*a[j+o+i]*w%MOD;
                Mod(a[j+o] = w1+w2);
                Mod(a[j+o+i] = w1+MOD-w2);
            }
        }
    }
    if(k==-1)
    {
        for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);
        int Inv = inv(len);
        for(int i=0;i<len;i++)a[i]=1ll*a[i]*Inv%MOD;
    }
}
int a[M],b[M],c[M];
int f[M],g[M],n;
void cdq(int l,int r)
{
    if(l==r)return ;
    int mid = (l+r)>>1;
    cdq(l,mid);
    get_lim(2*(r-l+1));
    for(int i=0;i<lim;i++)a[i]=b[i]=0;
    for(int i=0;i<=mid-l;i++)a[i]=f[l+i];
    for(int i=1;i<=r-l+1;i++)b[i]=g[i];
    ntt(a,lim,1),ntt(b,lim,1);
    for(int i=0;i<=lim;i++)c[i]=1ll*a[i]*b[i]%MOD;
    ntt(c,lim,-1);
    for(int i=mid+1-l;i<=r-l;i++)Mod(f[i+l]+=c[i]);
    cdq(mid+1,r);
}
int main()
{
//    freopen("tt.in","r",stdin);
    read(n);init(n<<1);f[0]=1;
    for(int i=1;i<n;i++)read(g[i]);
    cdq(0,lim-1);
    for(int i=0;i<n;i++)printf("%d ",f[i]);
    puts("");
    return 0;
}

原文地址:https://www.cnblogs.com/LiGuanlin1124/p/11162009.html

时间: 2024-08-01 06:12:00

分治FFT/NTT的相关文章

分治FFT/NTT 模板

题目要我们求$f[i]=\sum\limits_{j=1}^{i}f[i-j]g[j]\;mod\;998244353$ 直接上$NTT$肯定是不行的,我们不能利用尚未求得的项卷积 所以要用$CDQ$分治,先递归$[l,mid]$,然后处理$[l,mid]$对$[mid+1,r]$的影响,再递归$[mid+1,r]$ 当我们处理$[l,mid]$对$[mid+1,r]$的影响时,$f[i](i\in [l,mid])$的是已经求完的,所以能用$NTT$卷积 细节比较多,注意不要让$f[i](i\

2017 3 11 分治FFT

考试一道题的递推式为$$f[i]=\sum_{j=1}^{i} j^k \times (i-1)! \times \frac{f[i-j]}{(i-j)!}$$这显然是一个卷积的形式,但$f$需要由自己卷过来(我也不知到怎么说),以前只会生成函数的做法,但这题好像做不了(谁教教我怎么做),于是无奈的写了一发暴力,看题解发现是分治FFT.分治每层用$f[l]-f[mid]$与$a[1]-a[r-l]$做NTT.这样显然每个$f[l]-f[mid]$对$f[mid+1]-f[r]$的贡献都考虑到了.

【XSY2166】Hope 分治 FFT

题目描述 对于一个\(1\)到\(n\)的排列\(a_1,a_2,a_3,\ldots,a_n\),我们定义这个排列的\(P\)值和\(Q\)值: 对于每个\(a_i\),如果存在一个最小的\(j\)使得\(i<j\)且\(a_i<a_j\),那么将\(a_i\)和\(a_j\)连一条无向边.于是就得到一幅图.计算这幅图每个联通块的大小,将它们相乘,得到\(P\).记\(Q=P^k\). 对于\(1\)到\(n\)的所有排列,我们想知道它们的\(Q\)值之和.由于答案可能很大,请将答案对\(9

【XSY2744】信仰圣光 分治FFT 多项式exp 容斥原理

题目描述 有一个\(n\)个元素的置换,你要选择\(k\)个元素,问有多少种方案满足:对于每个轮换,你都选择了其中的一个元素. 对\(998244353\)取模. \(k\leq n\leq 152501\) 题解 吐槽 为什么一道FFT题要把\(n\)设为\(150000\)? 解法一 先把轮换拆出来. 直接DP. 设\(f_{i,j}\)为前\(i\)个轮换选择了\(j\)个元素,且每个轮换都选择了至少一个元素的方案数. \[ f_{i,j}=\sum_{k=1}^{a_i}f_{i-1,j

[bzoj3456]城市规划——分治FFT

题目大意: 求n个点的带标号简单无向联通图的数目. 思路: 嗯多项式求逆还不会,到时候会了应该会补吧. 这种和图计数有关的题目一般都是考虑反面计数或者是容斥什么的. 考虑枚举一号点的连通块的大小,然后用总方案数减去这些方案数. 可以得到递推式: \[ f_{i}=2^{i\choose 2}-\sum_{j=1}^{i-1}{i-1\choose j-1}\times f_{j}\times 2^{i-j\choose2} \] 后面的式子可以化为卷积的形式: \[ f_{i}=2^{i\cho

分治FFT

分治FFT 目的 解决这样一类式子: \[f[n] = \sum_{i = 0}^{n - 1}f[i]g[n - i]\] 算法 看上去跟普通卷积式子挺像的,但是由于计算\(f\)的每一项时都在利用它前面的项来产生贡献,所以不能一次FFT搞完.用FFT爆算复杂度\(O(n^2logn)\),比直接枚举复杂度还高-- 考虑优化这个算法,如果我们要计算区间\([l, r]\)内的\(f\)值,如果可以快速算出区间\([l, mid]\)内的\(f\)值对区间\([mid + 1, r]\)内的\(

分治FFT模板

题目链接:https://www.luogu.org/problemnew/show/P4721 总结了一下蒟蒻FFT/NTT容易写错的地方: ? 1.rev数组求错. ? 2.cdq注意顺序:先递归左, 处理左对右的影响,再递归右.(注意!这需要考虑到分治fft的原理!) ? 3.初始a数组忘了取模等各种忘取模. ? 4.NTT第二层循环i+=(1<<j)而不是i+=j ? 5.y=gnk*a[k+j]而不是a[k+j]. 接下来是AC代码 (打//标志的是曾经与现在本蒟蒻FFT写错的地方)

【UOJ#50】【UR #3】链式反应(分治FFT,动态规划)

[UOJ#50][UR #3]链式反应(分治FFT,动态规划) 题面 UOJ 题解 首先把题目意思捋一捋,大概就是有\(n\)个节点的一棵树,父亲的编号大于儿子. 满足一个点的儿子有\(2+c\)个,其中\(c\in A\),且\(c\)个儿子是叶子,另外\(2\)个存在子树,且两种点的链接的边是不同的,求方案数. 那么就考虑一个暴力\(dp\),设\(f[i]\)表示有\(i\)个节点的树的个数. 那么枚举它两个有子树的子树大小,然后把编号给取出来,得到: \[f[i]=\frac{1}{2}

【模板】分治 FFT

题目大意:给定长度为 \(n - 1\) 的序列 \(g\),求 \(f\) 序列,其中 \(f\) 为 \[ f[i]=\sum_{j=1}^{i} f[i-j] g[j] \] 学会了分治 \(fft\). 发现这个式子中也含有卷积,但是这是一个递推式,即:\(f\) 数组是未知的. 考虑分治策略,即:假设已经算出区间 \([l, mid]\) 的 \(f\) 值,现在要计算区间 \([mid + 1, r]\) 的 \(f\). 考虑左半部分对右半部分的贡献,对于 \[x \in [mid