hdu - 4710/4611 - Balls Rearrangement

题意：给出N, A, B，求

共T组测试数据(0 < T <= 50, 1 <= N <= 1000000000, 1 <= A,B <= 100000)。

题目链接(hdu 4710)：http://acm.hdu.edu.cn/showproblem.php?pid=4710

题目链接(hdu 4611)：http://acm.hdu.edu.cn/showproblem.php?pid=4611

——>>今天区域赛热身赛的第5题，看着别人一波波几分钟地AC，什么情况……这是。赛后才知道这是今年多校联合训练第二场第一题（hdu4611）～

找循环节——A和B的最小公倍数，计算表达式在循环节[0, lcm(A, B))上的和时，对于每一小段也一段段地统计。

注意溢出！

#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;

int N, A, B;

int gcd(int a, int b){
    return a % b == 0 ? b : gcd(b, a % b);
}

long long get(int L, int R){        //左闭右开
    long long ret = 0;      //要返回的结果
    int pa = 0, pb = 0;     //A和B里的指针
    for(int i = L; i < R;){     //从L开始
        long long toa = A - pa;       //需要toa次走完A末端
        long long tob = B - pb;       //需要tob次走完B末端
        long long tol = R - i;        //需要tol次走到R
        if(tol <= min(toa, tob)){       //当未走完末端已结束
            ret += abs(i % A - i % B) * tol;
            break;
        }
        if(toa < tob){      //当先走完A末端时
            pa = 0;     //走完A末端，回到最左端
            pb += toa;      //pb指针右移
        }
        else if(toa > tob){       //当先走完B末端时
            pb = 0;     //走完B末端，回到最左端
            pa += tob;      //pa指针右移
        }
        else pa = pb = 0;       //当同时走完各自的末端时，回到最左端
        ret += abs(i % A - i % B) * min(toa, tob);
        i += min(toa, tob);       //i右移
    }
    return ret;
}

void solve(){
    long long lcm = (long long)A / gcd(A, B) * B;       //求A，B的最小公倍数
    long long ret;
    if(N > lcm) ret = get(0, lcm) * (N / lcm) + get(N - N % lcm, N);
    else ret = get(0, N);
    printf("%I64d\n", ret);
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d%d", &N, &A, &B);
        solve();
    }
    return 0;
}