BZOJ 1257: [CQOI2007]余数之和sum

BZOJ 1257: [CQOI2007]余数之和sum

---

题目

Time Limit: 5 Sec Memory Limit: 162 MB

Description

给出正整数n和k，计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值，其中k mod i表示k除以i的余数。例如j(5, 3)=3 mod 1 + 3 mod 2 + 3 mod 3 + 3 mod 4 + 3 mod 5=0+1+0+3+3=7

Input

输入仅一行，包含两个整数n, k。

Output

输出仅一行，即j(n, k)。

Sample Input

5 3

Sample Output

7

HINT

50%的数据满足：1<=n, k<=1000 100%的数据满足：1<=n ,k<=10^9

---

题解

 m                        m
 ∑ n%i <=> n*m-∑ trunc(n/i)*i
i=1                      i=1

(数论渣。。。)

---

代码

#include<cstdio>
#include<cmath>
#define tt 1000000007
using namespace std;

long long n,m,ans,l,r;

int main()
{
    scanf("%lld%lld",&n,&m);
    for (int i=1;i<=trunc(sqrt(n));i++)
    {
        l=n/(i+1)+1;r=n/i;
        if (l>m) continue;
        if (r>=m) r=m;
        ans=(ans+n%tt*((r-l+1)%tt))%tt;
        if ((l+r)%2==0) ans=(ans-(l+r)/2%tt*((r-l+1)%tt)%tt*i)%tt;
         else ans=(ans-(l+r)%tt*((r-l+1)/2%tt)%tt*i)%tt;
        ans=(ans+tt)%tt;
    }
    for (int i=1;i<=n/(trunc(sqrt(n))+1);i++) if (i<=m) ans=(ans+n%i)%tt;
    if (m>n) ans=(ans+n%tt*((m-n)%tt))%tt;
    printf("%lld\n",ans);
    return 0;
}