[BZOJ2956]模积和（数论）

题目描述

传送门

题解

这题就是正儿八经的化（画）式（柿）子
首先计算$i=j$和$i\ne j$的总和
$ans1=\sum\limits_{i=1}^n\sum\limits_{j=1}^m(n\space mod\space i)*(m\space mod\space j)$
$=\sum\limits_{i=1}^n\sum\limits_{j=1}^m(n-{\lfloor{n\over i}\rfloor}i)*(m-{\lfloor{m\over j}\rfloor}j)$
$=\sum\limits_{i=1}^n\sum\limits_{j=1}^m(nm-n{\lfloor{m\over j}\rfloor}j-m{\lfloor{n\over i}\rfloor}i+{\lfloor{n\over i}\rfloor}i{\lfloor{m\over j}\rfloor}j)$
$=n^2m^2-n^2\sum\limits_{j=1}^m{\lfloor{m\over j}\rfloor}j-m^2\sum\limits_{i=1}^n{\lfloor{n\over i}\rfloor}i+\sum\limits_{i=1}^n{\lfloor{n\over i}\rfloor}i\sum\limits_{j=1}^m{\lfloor{m\over j}\rfloor}j$
然后计算$i=j$的情况
令$Min=min(n,m)$
$ans2=\sum\limits_{i=1}^{Min}(n\space mod\space i)*(m\space mod\space i)$
$=\sum\limits_{i=1}^{Min}(n-{\lfloor{n\over i}\rfloor}i)*(m-{\lfloor{m\over i}\rfloor}i)$
$=\sum\limits_{i=1}^{Min}(nm-n{\lfloor{m\over i}\rfloor}i-m{\lfloor{n\over i}\rfloor}i+{\lfloor{n\over i}\rfloor}{\lfloor{m\over i}\rfloor}i^2)$
$=nmMin-n\sum\limits_{i=1}^{Min}{\lfloor{m\over i}\rfloor}i-m\sum\limits_{i=1}^{Min}{\lfloor{n\over i}\rfloor}i+\sum\limits_{i=1}^{Min}{\lfloor{n\over i}\rfloor}{\lfloor{m\over i}\rfloor}i^2$
最后就是
$ans=ans1-ans2$

形如$\sum\limits_{i=1}^n{\lfloor{m\over i}\rfloor}i$的式子可以分块$O(\sqrt n)$来做，用到等差数列的求和公式
形如$\sum\limits_{i=1}^N{\lfloor{n\over i}\rfloor}{\lfloor{m\over i}\rfloor}i^2$的式子可以分块$O(\sqrt n+\sqrt m)$来做，用到平方数列的求和公式，其中二次方$\sum\limits_{i=1}^n i^2={n(n+1)(2n+1)\over 6}$

为什么叫做“膜”积和呢
因为代码写起来“膜”真 · tm多啊

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define Mod 19940417
#define inv 3323403

LL n,m,Min,ans1,ans2,ans;

LL s(LL n)
{
    LL ans=n*(2*n%Mod+1)%Mod*(n+1)%Mod*inv%Mod;
    return ans;
}
LL calc(LL n,LL k)
{
    LL ans=0;
    for (LL i=1,j=0;i<=n;i=j+1)
    {
        j=min(k/(k/i),n);
        ans+=(k/i)*(j-i+1)*(i+j)/2;
        ans%=Mod;
    }
    return ans;
}
LL CALC(LL N,LL n,LL m)
{
    LL ans=0;
    for (LL i=1,j=0;i<=N;i=j+1)
    {
        j=min(n/(n/i),m/(m/i));j=min(j,N);
        ans+=(n/i)%Mod*(m/i)%Mod*(s(j)-s(i-1))%Mod;
        ans%=Mod;
    }
    return ans;
}

int main()
{
    scanf("%lld%lld",&n,&m);
    ans1=n%Mod*n%Mod*m%Mod*m%Mod-n%Mod*n%Mod*calc(m,m)%Mod-m%Mod*m%Mod*calc(n,n)+calc(n,n)*calc(m,m)%Mod;
    ans1=(ans1%Mod+Mod)%Mod;
    Min=min(n,m);
    ans2=n%Mod*m%Mod*Min%Mod-n%Mod*calc(Min,m)%Mod-m%Mod*calc(Min,n)%Mod+CALC(Min,n,m);
    ans2=(ans2%Mod+Mod)%Mod;
    ans=(ans1-ans2+Mod)%Mod;
    printf("%lld\n",ans);
}