多项式开方板子

注意常数项要手动开方

void My_Sqrt(int a[],int Sqrt[],int toap)
{
    if(toap==1){Sqrt[0]=Get_Sqrt(a[0]);return;}
    My_Sqrt(a,Sqrt,toap>>1);
    fill(Sqrt+(toap>>1),Sqrt+(toap<<1),0);
    copy(a,a+toap,aa);
    fill(aa+toap,aa+(toap<<1),0);
    GetInv(Sqrt,sqrt_ni,toap);
    fill(sqrt_ni+toap,sqrt_ni+(toap<<1),0);
    NTT(sqrt_ni,toap<<1,1),NTT(aa,toap<<1,1),NTT(Sqrt,toap<<1,1);
    for(int i=0;i<(toap<<1);i++)Sqrt[i]=ch((MOD+1)>>1,add(Sqrt[i],ch(aa[i],sqrt_ni[i])));
    NTT(Sqrt,toap<<1,-1);
}

猜你喜欢

转载自blog.csdn.net/includelhc/article/details/80221260