分治NTT/在线卷积

https://www.luogu.com.cn/problem/P4721

已知 g g g,求在这里插入图片描述

考虑分治,现在在 [ l , r ] [l,r] [l,r],先计算 [ l , m i d ] [l, mid] [l,mid],然后计算 [ l , m i d ] [l, mid] [l,mid] [ m i d + 1 , r ] [mid+1,r] [mid+1,r] 的贡献。

分治NTT/在线卷积_第1张图片

计算左对右的贡献,就把左边的 f f f 拿出来,乘上 g g g,贡献到右边的 f f f 里面。

和普通dp的计算类似,只不过贡献变成了NTT而已。

void solve(int l, int r, int logn) {  //[l, r]
	if(!logn) return ; 
	int mid=(l+r)>>1; 
	solve(l, mid, logn-1); 
	int i, j; 
	for(i=0, j=l; j<=mid; ++i, ++j) a[i]=f[j]; // 取出左边的f
	for(; i<=(1<<logn); ++i) a[i]=0; 
	for(i=0; i<=(1<<logn); ++i) b[i]=g[i]; //拿出g
	revese(r-l+1, logn); 
	NTT(a, r-l+1, 1); NTT(b, r-l+1, 1); 
	for(i=0; i<r-l+1; ++i) a[i]=a[i]*b[i]%mo; //计算f*g
	NTT(a, r-l+1, -1); 
	for(i=mid+1, j=mid-l+1; i<=r; ++i, ++j) 
		f[i]=(f[i]+a[j])%mo; //贡献到右边
	solve(mid+1, r, logn-1); 	
}

你可能感兴趣的:(算法,分治NTT,NTT,在线卷积)