随便写点东西

2022/6/9 23:50:36

本文主要是介绍随便写点东西,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

学习多项式求逆的过程中,看着自己的代码怎么看怎么像是\(O(nlog^2n)\) 的,然后看到了大佬的写法:

void solve(LL *a,LL *b,int p){/*a:seq,b:inv*/
	if(p==1){
		b[0]=fpm(a[0],MOD-2);
		return;
	}
	solve(a,b,(p+1)>>1);
	lim=1; L=0;
	while(lim<(p<<1)){
		lim<<=1;
		L++;
	}
	for(int i=0;i<lim;i++)
		rev[i]=rev[i>>1]>>1|(1&i)<<(L-1);
	for(int i=0;i<p;i++)
		tmp[i]=a[i];
	for(int i=p;i<lim;i++)
		tmp[i]=0;
	NTT(tmp,1); NTT(b,1);
	for(int i=0;i<lim;i++)
		b[i]=(2-b[i]*tmp[i]%MOD+MOD)%MOD*b[i]%MOD;
	NTT(b,-1);
	for(int i=p;i<lim;i++)
		b[i]=0;
}

可以看到这份代码在每轮倍增时只做了两次NTT和一次INTT,关键就在于大佬没有按部就班地把递推式的每个乘积项都卷出来,而是直接用点值表达式做乘法,

这样一轮倍增的复杂度是\(O(len)\)的,其中len是当前\(b\)式的长度。

所以事实上NTT并不是每一轮都跑满,而是很多时候,卷积的两个式子都很短。

这就很妙了(不过可能因为我是傻逼才想不到吧)。

NTT又一次写萎了,哈哈,再次证明我是傻逼



这篇关于随便写点东西的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程