多项式全套(待续)

#include <stdio.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int P=333333,MOD=998244353;
inline int add(int a,int b)
{a+=b;return a>=MOD?a-MOD:a;}
inline int sub(int a,int b)
{a-=b;return a<0?a+MOD:a;}
inline int mul(int a,int b)
{return 1LL*a*b%MOD;}
inline int ksm(int a,int b)
{
	int ans=1;
	for(;b;b>>=1,a=mul(a,a))
		if(b&1)ans=mul(ans,a);
	return ans;
}
int w[P];
void NTT(int a[],int n,int f)
{
	for(int i=0,j=0;i<n;i++){
		if(i<j)swap(a[i],a[j]);
		for(int k=n>>1;(j^=k)<k;k>>=1);
	}
	w[0]=1;
	for(int m=1;m<n;m<<=1){
		int owo=ksm(3,(MOD-1)/(m<<1)*f+MOD-1);
		for(int j=1;j<m;j++)
			w[j]=mul(w[j-1],owo);
		for(int i=0;i<n;i+=(m<<1))
			for(int j=0;j<m;j++){
				int p=a[i+j],q=mul(a[i+j+m],w[j]);
				a[i+j]=add(p,q),a[i+j+m]=sub(p,q);
			}
	}
	if(f<0){
		int inv=ksm(n,MOD-2);
		for(int i=0;i<n;i++)
			a[i]=mul(a[i],inv);
	}
}
int mulb[P];
void Poly_Mul(int a[],int b[],int n)
{
	for(int i=0;i<n;i++)
		mulb[i]=b[i];
	NTT(a,n,1),NTT(mulb,n,1);
	for(int i=0;i<n;i++)
		a[i]=mul(a[i],mulb[i]);
	NTT(a,n,-1);
}
int inva[P];
void Poly_Inv(int a[],int inv[],int n)
{
	if(n==1){
		inv[0]=ksm(a[0],MOD-2);
		return;
	}
	Poly_Inv(a,inv,n>>1);
	fill(inv+(n>>1),inv+(n<<1),0);
	fill(inva+n,inva+(n<<1),0);
	for(int i=0;i<n;i++)
		inva[i]=a[i];
	NTT(inva,n<<1,1),NTT(inv,n<<1,1);
	for(int i=0;i<(n<<1);i++)
		inv[i]=mul(inv[i],sub(2,mul(inva[i],inv[i])));
	NTT(inv,n<<1,-1);
}
int sqq[P],sqa[P];
void Poly_Sqrt(int a[],int sq[],int n)
{
	if(n==1){
		//Not All
		sq[0]=1;
		return;
	}
	Poly_Sqrt(a,sq,n>>1);
	fill(sq+(n>>1),sq+n,0);
	Poly_Inv(sq,sqq,n);
	int maxn=1;
	while(maxn<=(n*3))maxn<<=1;
	for(int i=0;i<n;i++)
		sqa[i]=a[i];
	fill(sq+(n>>1),sq+maxn,0);
	fill(sqq+n,sqq+maxn,0);
	fill(sqa+n,sqa+maxn,0);
	NTT(sq,maxn,1);
	NTT(sqq,maxn,1);
	NTT(sqa,maxn,1);
	int i2=(MOD+1)/2;
	for(int i=0;i<maxn;i++)
		sq[i]=mul(mul(sqq[i],i2),add(sqa[i],mul(sq[i],sq[i])));
	NTT(sq,maxn,-1);
}
int moda[P],modb[P],modc[P];
void Poly_Mod(int a[],int b[],int na,int nb) 
{
	if(na<nb)return;
	reverse(a,a+na+1),reverse(b,b+nb+1);
	int n=1;
	while(n<=na-nb+1)n<<=1;
	Poly_Inv(b,modb,n);
	n<<=1;
	for(int i=0;i<=na-nb;i++) 
		moda[i]=a[i];
	fill(modb+na-nb+1,modb+n,0);
	fill(moda+na-nb+1,moda+n,0);
	NTT(modb,n,1),NTT(moda,n,1);
	for(int i=0;i<n;i++) 
		modc[i]=mul(moda[i],modb[i]);
	NTT(modc,n,-1);
	reverse(modc,modc+na-nb+1);
	reverse(a,a+na+1),reverse(b,b+nb+1);
	//
	for(n=1;n<=na;n<<=1);
	fill(modc+na-nb+1,modc+n,0);
	for(int i=0;i<=nb;i++) 
		modb[i]=b[i];
	fill(modb+nb+1,modb+n,0);
	NTT(modb,n,1),NTT(modc,n,1);
	for(int i=0;i<n;i++) 
		modb[i]=mul(modb[i],modc[i]);
	NTT(modb,n,-1);
	for(int i=0;i<=na;i++) 
		a[i]=sub(a[i],modb[i]);
}
int lna[P];
int inv[P];
void Poly_Ln(int a[],int r[],int n)
{
	for(int i=0;i<n;i++)
		r[i]=mul(a[i+1],i+1);
	fill(r+n,r+(n<<1),0);
	Poly_Inv(a,lna,n);
	fill(lna+n,lna+(n<<1),0);
	Poly_Mul(r,lna,n<<1);
	for(int i=n-1;i;--i)
		r[i]=mul(inv[i],r[i-1]);
	r[0]=0;
}
int expr[P],expa[P];
void Poly_Exp(int a[],int r[],int n)
{//r=r0*(1+a-lnr0)
	if(n==1){
		//r[0]=1;Maybe Useful
		return;
	}
	Poly_Exp(a,r,n>>1);
	fill(r+(n>>1),r+(n<<1),0);
	Poly_Ln(r,expr,n);
	for(int i=0;i<n;i++)
		expa[i]=sub(a[i],expr[i]);
	expa[0]=add(1,expa[0]);
	fill(expa+n,expa+(n<<1),0);
	Poly_Mul(r,expa,n<<1);
}
void Inv_Init(int maxn)
{
	inv[1]=1;
	for(int i=2;i<=maxn;i++)
		inv[i]=mul(inv[MOD%i],MOD-MOD/i);
}
int ksma[P];
void Poly_Ksm(int a[],int r[],int n,long long tmp)
{
	int mem=a[0],k=tmp%MOD;
	Poly_Ln(a,ksma,n);
	for(int i=0;i<n;i++)
		ksma[i]=mul(ksma[i],k);
	r[0]=ksm(mem,tmp%(MOD-1));
	Poly_Exp(ksma,r,n);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值