jzoj3192 【中山市选2013】球

求$\sum |i\mod a - i \mod b|$

分析！！！

1≤N≤1000000000，A,B≤100000
保证A < B先，否则互换。显然对答案无影响.

先给n扩到下一个 a的倍数. 最后再减掉多算的.
考虑模A=0的一堆数，由循环节可以统计出两个数组：
s[x]= %b=x的数的个数
s1[x]= %b=x的数的个数 * x

然后考虑%a余数k=0..A-1。
分绝对值是否翻转两类讨论：

假如说某个数之前模B=q，那么现在模B=(q+k)%b
1. 若不翻转，则该数x模B=0..k-1.
还原回%a余数等于0的情况： 往前挪k位，对应s中B-k..B-1
由预处理的s与s1，可以计算得到sum i%a与sum i%b
2. 若翻转，则该数x模B=k..B-1.
还原回%a余数等于0的情况： 往前挪k位，对应s中0,k-1
同上计算可得答案

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 100010;

typedef long long ll;
ll n,A,B,nt,ans;
ll s[N],s1[N],pre[N],pre1[N];
bool bz[N];
inline ll abs(ll x) {return x>0?x:-x;}
int main() {
    freopen("3192.in","r",stdin);
    cin>>nt>>A>>B; if (A>B) swap(A,B);
    n=nt+(A-(nt%A));
    for (ll i=0; i*A<n; i++) {
        ll yb=i*A%B;
        if (!bz[yb]) s[yb]++,bz[yb]=1;
        else {
            ll ti=(n/A) / i;
            for (int j=0; j<B; j++) s[j]*=ti;
            for (int j=ti*i; j*A<n; j++) s[j*A%B]++;
            break;
        }
    }
    pre[0]=s[0];
    for (ll i=1; i<B; i++) {
        s1[i]=s[i]*i;
        pre[i]=pre[i-1]+s[i],pre1[i]=pre1[i-1]+s1[i];
    }

    for (ll k=0; k<A; k++) {
        ll fj=B-k-1;
        ans+=pre1[fj];//+ k*pre[fj] - k*pre[fj];
        ans+=(pre[B-1] - pre[fj])*k - (pre1[B-1] - pre1[fj]-(B-k)*(pre[B-1] - pre[fj]));
    }

    for (ll i=nt; i<n; i++) ans-=abs(i%A-i%B);
    cout<<ans<<endl;
}