[BZOJ1257][CQOI2007]余数之和sum

1257: [CQOI2007]余数之和sum

Time Limit: 5 Sec  Memory Limit: 162 MB
Submit: 4604  Solved: 2143
[Submit][Status][Discuss]

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

 

 

x%y=x-(int)(x/y)*y

我们找出所有(int)(x/y)相等的y,按等差数列求和即可得出答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
long long n,k;
long long ans;
int main()
{
    scanf("%lld%lld",&n,&k);
    if(n>k) ans=(n-k)*k,n=k;
    for(long long i=1;i<=n;)
    {
        long long t=k/i,r=k/t;
        if(r>=n)r=n;
        ans+=(r-i+1)*k-(r-i+1)*(i+r)/2*t;
        i=r+1;
    }
    printf("%lld\n",ans);
    return 0;
}