洛谷 - P4721 【模板】分治 FFT(分治NTT)

题目链接：点击查看

题目大意：给出序列 g1,⋯,ng_{1,\cdots,n}g1,⋯,n，求 f0,⋯,nf_{0,\cdots,n}f0,⋯,n

规定 fi=∑j=1ifi−jgjf_i=\sum\limits_{j=1}^{i}f_{i-j}g_jfi=j=1∑ifi−jgj，其中 f0=1f_0=1f0=1

题目分析：假如将区间一分为二，不难发现左侧的区间会对右侧的区间提供贡献，所以我们不妨参考cdq分治的思路，先将左侧区间都算出答案，然后再递归进入右侧区间，每次将数组偏移一下然后卷积就好啦

代码：

// Problem: P4721 【模板】分治 FFT
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P4721
// Memory Limit: 125 MB
// Time Limit: 1000 ms
//
// Powered by CP Editor (https://cpeditor.org)// #pragma GCC optimize(2)
// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<list>
#include<unordered_map>
#define lowbit(x) (x&-x)
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
template<typename T>
inline void read(T &x)
{T f=1;x=0;char ch=getchar();while(0==isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(0!=isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();x*=f;
}
template<typename T>
inline void write(T x)
{if(x<0){x=~(x-1);putchar('-');}if(x>9)write(x/10);putchar(x%10+'0');
}
const int inf=0x3f3f3f3f;
const int md=998244353,N=5e6+100,g3=(md+1)/3;
int f[N],g[N],a[N],b[N];
namespace poly{int lim,rev[N],inv[N];inline void upd(int&a){a+=a>>31&md;}inline int pow(int a,int b){int r=1;for(;b;b>>=1,a=(LL)a*a%md)if(b&1)r=(LL)r*a%md;return r;}inline int poly_start(){inv[1]=1;for(int i=2;i<N;++i)inv[i]=(md-md/i)*(LL)inv[md%i]%md;return 1;}int __START__=poly_start();inline void init(int n){int l=-1;for(lim=1;lim<n;lim<<=1)++l;for(int i=1;i<lim;++i)rev[i]=(rev[i>>1]>>1)|((i&1)<<l);}void NTT(int*a,int f){for(int i=1;i<lim;++i)if(i<rev[i])std::swap(a[i],a[rev[i]]);for(int i=1;i<lim;i<<=1){const int gi=pow(f?3:g3,(md-1)/(i<<1));for(int j=0;j<lim;j+=i<<1)for(int k=0,g=1;k<i;++k,g=(LL)g*gi%md){const int x=a[j+k],y=a[j+k+i]*(LL)g%md;upd(a[j+k]+=y-md),upd(a[j+k+i]=x-y);}}if(!f){const LL iv=inv[lim];for(int i=0;i<lim;++i)a[i]=a[i]*iv%md;}}void MUL(int *a,int n){init(n);static int P[N],Q[N];for(int i=0;i<n;i++) P[i]=a[i];for(int i=n;i<lim;i++) P[i]=0;for(int i=0;i<lim;i++) Q[i]=g[i];NTT(P,1);NTT(Q,1);for(int i=0;i<lim;i++) P[i]=(1LL*P[i]*Q[i])%md;NTT(P,0);for(int i=0;i<lim;i++) a[i]=P[i];}
}
void solve(int l,int r) {if(l==r) {if(!l) {f[l]=1;}return;}int mid=(l+r)>>1;solve(l,mid);for(int i=l;i<=mid;i++) a[i-l]=f[i];for(int i=mid+1;i<=r;i++) a[i-l]=0;poly::MUL(a,r-l+1);for(int i=mid+1;i<=r;i++) f[i]=(f[i]+a[i-l])%md;solve(mid+1,r);
}
int main()
{#ifndef ONLINE_JUDGE
//  freopen("data.in.txt","r",stdin);
//  freopen("data.out.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);int n;read(n);for(int i=1;i<n;i++) {read(g[i]);}solve(0,n-1);for(int i=0;i<n;i++) {printf("%d ",f[i]);}return 0;
}

洛谷 - P4721 【模板】分治 FFT(分治NTT)相关推荐

[洛谷P4721]【模板】分治 FFT
题目大意:给定长度为$n-1$的数组$g_{[1,n)}$,求$f_{[0,n)}$,要求: $$ f_i=\sum_{j=1}^if_{i-j}g_j\\ f_0=1 $$ 题解:直接求复杂度是$O ...
洛谷·【模板】点分树 | 震波【including 点分树
初见安-这里是传送门:洛谷P6329 [模板]点分树 | 震波一.点分树其实你会点分治的话,点分树就是把点分治时的重心提出来重新连城一棵树. 比如当前点是u,求出子树v的重心root后将root与 ...
python高精度乘法_洛谷P1919--A*B Problem升级版（NTT优化高精度乘法）
题目背景本题数据已加强,请使用 FFT/NTT,不要再交 Python 代码浪费评测资源. 题目描述给你两个正整数 a,b,求$ a \times b$. 输入格式第一行一个正整数,表示 a: ...
专题·树链剖分【including 洛谷·【模板】树链剖分
初见安~~~终于学会了树剖~~~ [兴奋]当初机房的大佬在学树剖的时候我反复强调过:"学树剖没有前途的!!!" 恩.真香. 一.重链与重儿子所谓树剖--树链剖分,就是赋予一个链的 ...
洛谷P4233 射命丸文的笔记分治NTT+竞赛图
Description 给定n 对于i从1~n,输出i个点组成竞赛图中,哈密顿回路的平均数量 Solution 竞赛图存在哈密顿回路的充要条件就是强连通设f(n)f(n)f(n)表示n个点形成强连通 ...
P4721-[模板]分治FFT【NTT,分治】
正题题目链接:https://www.luogu.com.cn/problem/P4721 题目大意给出序列ggg,然后f0=1f_0=1f0=1 fi=∑j=1ifi−jgjf_{i}=\su ...
洛谷 P1919 模板】A*B Problem升级版（FFT快速傅里叶）
https://www.luogu.com.cn/problem/P1919 题目背景本题数据已加强,请使用 FFT/NTT,不要再交 Python 代码浪费评测资源. 题目描述给你两个正整数 a ...
洛谷.4245.[模板]任意模数NTT(MTT/三模数NTT)
题目链接三模数$NTT$: 就是多模数$NTT$最后$CRT$一下...下面两篇讲的都挺明白的. https://blog.csdn.net/kscla/article/details/ ...
[洛谷P1498] 南蛮图腾（分治+递归）
分析根据题目标签,首先解释一下分治算法:分治算法的基本思想是将一个规模为N的问题分解为K个规模较小的子问题,这些子问题相互独立且与原问题性质相同.求出子问题的解,就可得到原问题的解.即一种分目标完成 ...

洛谷 - P4721 【模板】分治 FFT(分治NTT)

洛谷 - P4721 【模板】分治 FFT(分治NTT)相关推荐

最新文章

热门文章