hdu 4710 Balls Rearrangement 数论

这个公倍数以后是循环的很容易找出来，然后循环以内的计算是打表找的规律，规律比较难表述，自己看代码吧。。

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
long long n,a,b;
long long lcd(int a,int b)
{
    long long ret=a;
    ret*=b;
    int tmp;
    while(a%b)
    {
        tmp=a%b;
        a=b;
        b=tmp;
    }
    return ret/b;
}
long long cal(long long tmp)
{
//    cout<<tmp<<endl;
    long long sum=0,ret,i;
//    for(i=0;i<tmp;i++)
//    sum+=fabs(i%a-i%b);
//    return sum;
    for(i=0;i+a-1<tmp;i+=a)
    {
        ret=b-(i%b);
        if(ret>=a)
        sum+=(i%b)*a;
        else
        sum+=ret*(i%b)+(a-ret)*(b-(i%b));
    }
    for(;i<tmp;i++)
    sum+=fabs(i%a-i%b);
    return sum;
}
int main()
{
//    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld %lld %lld",&n,&a,&b);
        if(a>b) swap(a,b);
        long long m=lcd(a,b);
        long long ans=cal(m);
        ans*=n/m;
        ans+=cal(n%m);
        cout<<ans<<endl;
    }
    return 0;
}