bzoj2956 模积和（分块（平方的前缀和公式））

Description

求∑∑((n mod i)*(m mod j))其中1<=i<=n,1<=j<=m,i≠j。

Input

第一行两个数n,m。

Output

一个整数表示答案mod 19940417的值

Sample Input

3 4

Sample Output

1

样例说明
　　答案为(3 mod 1) * (4 mod 2)+(3 mod 1) * (4 mod 3)+(3 mod 1) * (4 mod 4) + (3 mod 2) * (4 mod 1) + (3 mod 2) * (4 mod 3) + (3 mod 2) * (4 mod 4) + (3 mod 3) * (4 mod 1) + (3 mod 3) * (4 mod 2) + (3 mod 3) * (4 mod 4) = 1

数据规模和约定

　　对于100%的数据n,m<=10^9。


分析：
这道题并没有用到反演，但是分块的方法和反演中一样

计算中会用到bzoj1257的奇技淫巧

在计算的时候，唯一有点难度的就是i^2的求和
我们给出公式：

//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long

using namespace std;

const ll p=19940417;
const ll ni=3323403;      //6的逆元
int n,m;

ll cal(int n)
{
    ll ans=0;
    int last;
    for (int i=1;i<=n;i=last+1)
    {
        last=n/(n/i);
        ll N=(ll)((n*(last-i+1)%p-(n/i)*(last-i+1)*(i+last)/2%p)%p+p)%p;
        ans=(ans+N%p)%p;
    }
    return ans%p;
}

ll sum(int x)
{
    return (ll)x*(x+1)%p*(2*x+1)%p*ni%p;
}

ll solve()
{
    ll ans=0;
    int last;
    for (int i=1;i<=min(n,m);i=last+1)
    {
        last=min(n/(n/i),m/(m/i));
        ll t1=(ll)n*m%p*(last-i+1)%p;
        ll t2=(ll)(n/i*m+m/i*n)%p*(last-i+1)*(i+last)/2%p;
        ll t3=(ll)(n/i)*(m/i)%p*(sum(last)-sum(i-1)+p)%p;
        ans=((ans+t1-t2+t3)%p+p)%p;
    }
    return ans%p; 
}

int main()
{
    scanf("%d%d",&n,&m);
    ll ans;
    ans=cal(n)*cal(m);
    printf("%lld",((ans-solve())%p+p)%p);
    return 0;
}