多项式开根

B^2=A

2ln(B)=ln(A)

ln(B)=\frac{ln(A)}{2}

B=e^{\frac{ln(A)}{2}}

那么先多项式ln,乘2的逆元,再exp即可。

代码:

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar(); 
	}return cnt*f;
}
const int mod=998244353;
int a[2000003],b[2000030],c[2000003],d[2000003],e[2000003],f[2000003],g[2000003],h[2000003],r[2000003],limit,l;
int inv2;
void getl(int len){
	limit=1,l=0;
	while(limit<len)limit<<=1,l++;
	for(int i=0;i<limit;i++){
		r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	}
}
int ksm(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=ans*a%mod;
		a=a*a%mod;b>>=1;
	}return ans;
}
void NTT(int *a,int len,int inv){
	for(int i=0;i<len;i++){
		if(i<r[i])swap(a[i],a[r[i]]);
	}
	//for(int i=0;i<len;i++)cout<<a[i]<<" ";cout<<endl; 
	for(int mid=1;mid<len;mid<<=1){
		int wn=ksm(3,(mod-1)/(mid<<1));
		for(int i=0;i<len;i+=mid*2){
			int omega=1;
			for(int j=0;j<mid;j++,omega=omega*wn%mod){
				int x=a[i+j],y=a[i+j+mid]*omega%mod;
				a[i+j]=(x+y)%mod;a[i+j+mid]=(x-y+mod)%mod;
			}
		}
	}//cout<<"#";
	if(inv==1)return;
	int gu=ksm(len,mod-2);reverse(a+1,a+len);
	for(int i=0;i<len;i++)a[i]=a[i]*gu%mod;
} 
void qiudao(int *a,int *b,int len){
	for(int i=1;i<len;i++)b[i-1]=a[i]*i%mod;b[len-1]=0;
}
void shita(int *a,int *b,int len){
	for(int i=1;i<len;i++)b[i]=a[i-1]*ksm(i,mod-2)%mod;b[0]=0;
}
void getinv(int *a,int *d,int len){
	if(len==1){
		d[0]=ksm(a[0],mod-2);return;
	}getinv(a,d,(len+1)>>1);
	getl(len<<1);
	for(int i=0;i<limit;i++)if(i<len)e[i]=a[i],f[i]=d[i];else e[i]=f[i]=0;
	//for(int i=0;i<len;i++)cout<<e[i]<<" ";cout<<endl;
	NTT(e,limit,1);NTT(f,limit,1);
	for(int i=0;i<limit;i++)d[i]=f[i]*(2-e[i]*f[i]%mod+mod)%mod;
	NTT(d,limit,-1);for(int i=len;i<limit;i++)d[i]=0;
}
void getln(int *a,int *b,int len){
	qiudao(a,c,len);getinv(a,d,len);
	//for(int i=0;i<len;i++)cout<<d[i]<<" ";cout<<endl;
	getl(len<<1);
	for(int i=0;i<limit;i++){
		e[i]=i<len?c[i]:0;
		f[i]=i<len?d[i]:0; 
	}
	NTT(e,limit,1);NTT(f,limit,1);
	for(int i=0;i<limit;i++)e[i]=e[i]*f[i]%mod;
	NTT(e,limit,-1);shita(e,b,len);
}
int tl[2000003],tr[2000003];
void getexp(int *a,int *b,int len){
	if(len==1){
		b[0]=1;return;
	}getexp(a,b,(len+1)>>1);getln(b,c,len);
	getl(len<<1);
	for(int i=0;i<limit;i++){
		if(i<len)tl[i]=b[i],tr[i]=(a[i]-c[i]+mod)%mod;
		else tl[i]=tr[i]=0;
	}tr[0]++;
	NTT(tl,limit,1);NTT(tr,limit,1);
	for(int i=0;i<limit;i++)b[i]=tl[i]*tr[i];
	NTT(b,limit,-1);for(int i=len;i<limit;i++)b[i]=0;
}
signed main(){
	inv2=ksm(2,mod-2);
	int n=in;int m;
	for(m=1;m<=n;m<<=1);
	for(int i=0;i<n;i++)a[i]=in;
	getln(a,b,m);//b[n-1]%=mod;
	//for(int i=0;i<n;i++)cout<<b[i]<<" ";cout<<endl;
	for(int i=0;i<m;i++)a[i]=b[i]%mod*inv2%mod,b[i]=0;
	getexp(a,b,m);
	for(int i=0;i<n;i++)cout<<b[i]<<" ";	
	
	return 0;
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值