hdu - 4710/4611 - Balls Rearrangement

来源：互联网发布：arm体系结构与编程编辑：程序博客网时间：2024/06/05 04:30

题意：给出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;}