萌新求助,分治乘法就 TLE 了/dk
  • 板块P4705 玩游戏
  • 楼主Provicy
  • 当前回复6
  • 已保存回复6
  • 发布时间2021/5/3 15:56
  • 上次更新2023/11/4 23:47:39
查看原帖
萌新求助,分治乘法就 TLE 了/dk
98618
Provicy楼主2021/5/3 15:56

RT,为什么跑这么慢呢/kel (Solve 函数)

#include <bits/stdc++.h>
#pragma GCC optimize(3)
//#define int long long
#define ri register
#define mk make_pair
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define is insert
#define es erase
#define vi vector<int>
#define vpi vector<pair<int,int>>
using namespace std; const int N=550010, Mod=998244353;
inline int read()
{
	int s=0, w=1; ri char ch=getchar();
	while(ch<'0'||ch>'9') { if(ch=='-') w=-1; ch=getchar(); }
	while(ch>='0'&&ch<='9') s=(s<<3)+(s<<1)+(ch^48), ch=getchar();
	return s*w;
}
int n,m,K,fac[N+5],inv[N+5];
vector<int> a,b;
int rev[N],r[24][2];
inline int ksc(int x,int p) { int res=1; for(;p;p>>=1, x=1ll*x*x%Mod) if(p&1) res=1ll*res*x%Mod; return res; }
inline void Get_Rev(int T) { for(ri int i=0;i<T;i++) rev[i]=(rev[i>>1]>>1)|((i&1)?(T>>1):0); }
inline void DFT(int T,vector<int> &s,int type)
{
	for(ri int i=0;i<T;i++) if(rev[i]<i) swap(s[i],s[rev[i]]);
	for(ri int i=2,cnt=1;i<=T;i<<=1,cnt++)
	{
		int wn=r[cnt][type];
		for(ri int j=0,mid=(i>>1);j<T;j+=i)
		{
			for(ri int k=0,w=1;k<mid;k++,w=1ll*w*wn%Mod)
			{
				int x=s[j+k], y=1ll*w*s[j+mid+k]%Mod;
				s[j+k]=x+y;
				if(s[j+k]>=Mod) s[j+k]-=Mod;
				s[j+mid+k]=x-y;
				if(s[j+mid+k]<0) s[j+mid+k]+=Mod;
			}
		}
	}
	if(!type) for(ri int i=0,inv=ksc(T,Mod-2);i<T;i++) s[i]=1ll*s[i]*inv%Mod;
}
inline void NTT(int n,int m,vector<int> &A,vector<int> B)
{
	int len=n+m;
	int T=1;
	while(T<=len) T<<=1;
	Get_Rev(T);
	A.resize(T), B.resize(T);
	DFT(T,A,1), DFT(T,B,1);
	for(ri int i=0;i<T;i++) A[i]=1ll*A[i]*B[i]%Mod;
	DFT(T,A,0);
}
void GetInv(int n,vector<int> &F,vector<int> G)
{
	if(n==1) { F[0]=ksc(G[0],Mod-2); return; }
	GetInv((n+1)/2,F,G);
	vector<int> A,B;
	int T=1;
	while(T<=n+n) T<<=1;
	Get_Rev(T);
	A.resize(T), B.resize(T);
	for(ri int i=0;i<n;i++) A[i]=F[i], B[i]=G[i];
	DFT(T,A,1), DFT(T,B,1);
	for(ri int i=0;i<T;i++) A[i]=(2ll*A[i]%Mod-1ll*B[i]*A[i]%Mod*A[i]%Mod+Mod)%Mod;
	DFT(T,A,0);
	for(ri int i=0;i<n;i++) F[i]=A[i];
}
inline void GetDao(int n,vector<int> &A,vector<int> B)
{
	for(ri int i=0;i<n-1;i++) A[i]=1ll*B[i+1]*(i+1)%Mod;
	A[n-1]=0;
}
inline void GetJi(int n,vector<int> &A,vector<int> B)
{
	for(ri int i=1;i<n;i++) A[i]=1ll*B[i-1]*inv[i]%Mod*fac[i-1]%Mod;
	A[0]=0;
}
inline void GetLn(int n,vector<int> &F,vector<int> G)
{
	vector<int> A,B;
	A.resize(n), B.resize(n);
	GetDao(n,A,G);
	GetInv(n,B,G);
	NTT(n,n,A,B);
	GetJi(n,F,A);
}
void Solve(int l,int r,vector<int> &F,vector<int> G)
{
	if(l==r)
	{
		F.resize(2);
		F[0]=1, F[1]=Mod-G[l];
		return;
	}
	int mid=(l+r)/2;
	Solve(l,mid,F,G);
	vector<int> C;
	Solve(mid+1,r,C,G);
	NTT(F.size()-1,C.size()-1,F,C);
}
signed main()
{
	r[23][1]=ksc(3,119), r[23][0]=ksc(ksc(3,Mod-2),119);
	for(ri int i=22;~i;i--) r[i][0]=1ll*r[i+1][0]*r[i+1][0]%Mod, r[i][1]=1ll*r[i+1][1]*r[i+1][1]%Mod;
	fac[0]=1;
	for(ri int i=1;i<=N;i++) fac[i]=1ll*fac[i-1]*i%Mod;
	inv[N]=ksc(fac[N],Mod-2);
	for(ri int i=N;i;i--) inv[i-1]=1ll*inv[i]*i%Mod;
	n=read(), m=read();
	a.resize(n+1), b.resize(m+1);
	for(ri int i=1;i<=n;i++) a[i]=read();
	for(ri int i=1;i<=m;i++) b[i]=read();
	K=read(), K++;
	vector<int> A,B;
	A.resize(n+1), B.resize(m+1);
	Solve(1,n,A,a), Solve(1,m,B,b);
	if((int)A.size()>=K) A.erase(A.begin()+K,A.end());
	if((int)B.size()>=K) B.erase(B.begin()+K,B.end());
	A.resize(K), B.resize(K);
	GetLn(K,A,A), GetLn(K,B,B);
	GetDao(K,A,A), GetDao(K,B,B);
	for(ri int i=K-1;i;i--) A[i]=Mod-A[i-1]; A[0]=n;
	for(ri int i=K-1;i;i--) B[i]=Mod-B[i-1]; B[0]=m;
	for(ri int i=0;i<K;i++) A[i]=1ll*A[i]*inv[i]%Mod, B[i]=1ll*B[i]*inv[i]%Mod;
	NTT(K,K,A,B);
	for(ri int i=1,iv=ksc(1ll*n*m%Mod,Mod-2);i<K;i++)
	{
		printf("%d\n",1ll*fac[i]*iv%Mod*A[i]%Mod)%Mod;
	}
	return 0;
}
2021/5/3 15:56
加载中...