hdu 4611 Balls Rearrangement

http://acm.hdu.edu.cn/showproblem.php?pid=4611

从A中向B中移动和从B中向A中移动的效果是一样的，我们假设从B中向A中移动 而且A>B

我们先求出所有在B[0]上的点移动到A上的分布情况 可以求出花费

当我们要求B[1]上的点移动到A上的分布情况时 相当于B[0]在A上的分布情况水平右移一个单位

由于B[1]点对于B[0]也向后移动了一个单位 所有相对位置没有移动

但是A中有一个地方需要改动 那就是每次A中最后一个位置的点需要移动到A[0]位置(因为对A进行取模的原因)

还有一个要注意的情况就是 B 在向后移动的时候 点的个数可能会减小 要特殊处理一下

代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<vector>
#include<algorithm>
#include<queue>
#include<bitset>
#include<deque>
#include<numeric>

//#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

typedef long long ll;
typedef unsigned int uint;
typedef pair<int,int> pp;
const double eps=1e-9;
const int INF=0x3f3f3f3f;
const ll MOD=1000000007;
const int H=210000;
const int K=100001;
ll a[H];
int f(int x)
{
    return x+K;
}
ll readya(ll N,ll A,ll B)
{
    ll lcm=A*B/(__gcd(A,B));
    int n=lcm/B;
    int k=0;
    ll num=(N/B);
    if(N%B) ++num;
    for(int i=0;i<n;++i)
    {
        a[f(k)]+=num/n;
        k=(k+B)%A;
    }
    ll M=num%n;
    k=0;
    while(M--)
    {
        a[f(k)]++;
        k=(k+B)%A;
    }
    ll tmp=0;
    for(int i=1;i<A;++i)
    tmp+=a[f(i)]*i;
    return tmp;
}
int main()
{
    //freopen("data.in","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(a,0,sizeof(a));
        ll N,A,B;
        cin>>N>>A>>B;
        if(B>A) swap(A,B);
        if(A==B)
        {printf("0\n");continue;}
        ll L=0;
        ll R=readya(N,A,B);
        int num=N/B;
        if(N%B) ++num;
        ll ans=L+R;
        for(int i=1;i<B;++i)
        {
            a[f(-i)]=a[f(A-i)];
            R-=(a[f(A-i)]*(A-i));
            L+=(a[f(-i)]*i);
            a[f(A-i)]=0;
            int tmp=(N-i)/B;
            if((N-i)%B) ++tmp;
            if(tmp!=num)
            {
                int x=tmp*B+i;
                a[f(x%A-i)]--;
                if(x%A-i>0) {R-=abs(x%A-i);}
                if(x%A-i<0) {L-=abs(x%A-i);}
                num=tmp;
            }
            ans+=(L+R);
        }
        cout<<ans<<endl;
    }
    return 0;
}