题目地址: http://acm.hdu.edu.cn/showproblem.php?pid=4611
题解:显然一个个求和时会以lcm(a,b)为周期,但是一旦a,b公因子比较少,这样也还是会超时的, i++这个跨度太小. 注意到如果mod a的序列在上升,mod b的序列也在上升,则ans+= 的那个值是不变的,这样我们就可以增加跨度了。 考虑从i开始的跨度为jump的序列时,只需要去 a-i%a ,b-i%b的较小值就可以了。 最后注意一下i+jump不可以超过边界,一旦超过,我们把i赋值成很大,这样保证会退出去就可以了
#include<iostream>
#include<cmath>
using namespace std;
typedef long long inta;
inta min(inta a,inta b)
{
if(a>b) return b;
else return a;
}
inta gcd(inta a,inta b)
{
if(b==0) return a;
else return gcd(b,a%b);
}
inta lcm(inta a,inta b)
{
return a/gcd(a,b)*b;
}
inta cost(inta a,inta b,inta n)
{
inta ans=0;
for(int i=0;i<n;)
{
inta jump=min(a-i%a,b-i%b);
// 边界处理
if(i+jump>=n+1)
{
jump=n-i;
i=n;
}
ans+=(jump)*fabs(i%a-i%b);
//last=i;
i+=jump;
}
return ans;
}
int main()
{
int T;
cin>>T;
while(T--)
{
inta n,a,b;
cin>>n>>a>>b;
inta ans=0;
inta q=0;
if(n>=lcm(a,b))
{
q=n/lcm(a,b);
ans=q*cost(a,b,lcm(a,b));
}
ans+=cost(a,b,n-q*lcm(a,b));
cout<<ans<<endl;
}
}