P5435 基于值域预处理的快速 GCD TLE 20pts求调
  • 板块学术版
  • 楼主OneLeft
  • 当前回复2
  • 已保存回复2
  • 发布时间2024/8/5 08:39
  • 上次更新2024/8/5 09:48:54
查看原帖
P5435 基于值域预处理的快速 GCD TLE 20pts求调
574215
OneLeft楼主2024/8/5 08:39

rt

#include<bits/stdc++.h>
using namespace std;
const int N=5005,M=1e6+5,K=1e3+5,mod=998244353;
int gcd(int x,int y){return (y==0?x:gcd(y,x%y));}
long long fastpow(int n,int k)
{
	long long res=1,base=n;
	for(;k>0;k/=2)
	{
		if(k%2==1)res=res*base%mod;
		base=base*base%mod;
	}
	return res;
}
bool flag[M];
int fac[M][5],Gcd[K][K],tot,prime[M];
void init(int n)
{
	fac[1][1]=fac[1][2]=fac[1][3]=1;
	for(int i=2;i<=n;i++)
	{
		if(!flag[i])
		{
			fac[i][1]=fac[i][2]=1,fac[i][3]=i;
			prime[++tot]=i;
		}
		for(int j=1;j<=tot,prime[j]*i<=n;j++)
		{
			int tmp=prime[j]*i;
			flag[tmp]=true;
			fac[tmp][1]=fac[i][1]*prime[j];
			fac[tmp][2]=fac[i][2];
			fac[tmp][3]=fac[i][3];
			if(fac[tmp][1]>fac[tmp][2])std::swap(fac[tmp][1],fac[tmp][2]);
			if(fac[tmp][2]>fac[tmp][3])std::swap(fac[tmp][2],fac[tmp][3]);
			if(i%prime[j]==0)break;
		}
	}
	for(int i=0;i*i<=n;i++)
	for(int j=0;j*j<=n;j++)
		Gcd[i][j]=gcd(i,j);
}
int query(int x,int y)
{
	int res=1;
	for(int i=1;i<=3;i++)
	{
		int r;
		if(fac[x][i]>1e3)r=(y%fac[x][i]==0?fac[x][i]:1);
		else r=Gcd[fac[x][i]][y%fac[x][i]];
		y/=r,res*=r;
	}
	return res;
}
int n,a[N],b[N];
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	init(1e6);
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)cin>>b[i];
	for(int i=1;i<=n;i++)
	{
		long long ans=0;
		for(int j=1;j<=n;j++)ans=(ans+fastpow(i,j)*query(a[i],b[j]))%mod;
		cout<<ans<<'\n';
	}
	return 0;
}
2024/8/5 08:39
加载中...