有两个坑点:
1.给你的a[i]可能是负数,用a[i]=(a[i]%m[i]+m[i])%m[i]。
2.最后a[i]*tp*t[i]会爆long long,用快速乘
普通代码:
#include<bits/stdc++.h>
using namespace std;
int k,a[15],m[15];
void exgcd(int a,int b,int &x,int &y)
{
if(b==0){ x=1; y=0; return;}
exgcd(b,a%b,x,y);
int tp=x;
x=y; y=tp-a/b*y;
}
int china()
{
int ans=0,lcm=1,x,y;
for(int i=1;i<=k;++i) lcm*=m[i];
for(int i=1;i<=k;++i)
{
int tp=lcm/m[i];
exgcd(tp,m[i],x,y);
x=(x%m[i]+m[i])%m[i];//x要为最小非负整数解
ans=(ans+tp*x*a[i])%lcm;
}
return (ans+lcm)%lcm;
}
int main()
{
scanf("%d",&k);
for(int i=1;i<=k;i++)
scanf("%d",&a[i]);
for(int i=1;i<=k;i++)
scanf("%d",&m[i]);
printf("%d\n",china());
return 0;
}
AC代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll k,a[15],m[15];
void exgcd(ll a,ll b,ll &x,ll &y)
{
if(b==0){ x=1; y=0; return;}
exgcd(b,a%b,x,y);
ll tp=x;
x=y; y=tp-a/b*y;
}
ll mul(ll a,ll b,ll mod)
{
ll res=0;
while(b)
{
if(b&1)res=(res+a)%mod;
a=(a+a)%mod;
b>>=1;
}
return res;
}
ll china()
{
ll ans=0,lcm=1,x,y;
for(int i=1;i<=k;++i) lcm*=m[i];
for(int i=1;i<=k;++i)
{
ll tp=lcm/m[i];
exgcd(tp,m[i],x,y);
x=(x%m[i]+m[i])%m[i];//x要为最小非负整数解
ans=(ans+mul(mul(tp,x,lcm),a[i],lcm))%lcm;
}
return (ans+lcm)%lcm;
}
int main()
{
scanf("%d",&k);
for(int i=1;i<=k;i++)
scanf("%lld",&a[i]);
for(int i=1;i<=k;i++)
scanf("%lld",&m[i]);
for(int i=1;i<=k;i++) a[i]=(a[i]%m[i]+m[i])%m[i];
printf("%lld\n",china());
return 0;
}