199. 余数之和(数论--分块)

Problem Link：https://www.acwing.com/problem/content/description/201/

 

给出正整数n和k，计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值。

例如j(5, 3)=3 mod 1 + 3 mod 2 + 3 mod 3 + 3 mod 4 + 3 mod 5=0+1+0+3+3=7。

输入格式

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

输出格式

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

数据范围

1≤n,k≤1091≤n,k≤109

输入样例：

5 3

输出样例：

7

 

 

AC code：

 

#include<bits/stdc++.h>
#define LL long long
using namespace std;

LL n,k,ans;

int main()
{
    scanf("%lld%lld",&n,&k);
    ans=n*k;
    int last;
    for(int i=1;i<=n;i=last+1)
    {
    	//last=k/i!=0?min(k/(k/i),n):n;
        if(i<=k)
        {
            last=min(n,k/(k/i));
        }
        else
        {
            last=n;
        }
        //ans-=(i+last)*(last-i+1)/2*(k/i);注意这样写WA！！！ 
        ans-=(k/i)*(i+last)*(last-i+1)/2;
    }
    printf("%lld\n",ans);
    return 0;
}