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;
}