【bzoj 1257】[CQOI2007]余数之和sum

1257: [CQOI2007]余数之和sum

Time Limit: 5 Sec Memory Limit: 162 Mb Submit: 3708 Solved: 1704
[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

Source
[Submit][Status][Discuss]

【题解】

∑i=1nkmodi

=∑i=1nk−⌊ki⌋∗i

=n∗k−∑i=1n⌊ki⌋∗i

当i>k时，⌊ki⌋∗i=0

so,只需考虑i≤k的情况，即：

=n∗k−∑i=1min(n,k)⌊ki⌋∗i

根据性质可知：⌊ki⌋的取值不超过k√个连续区间

用i从1到min(n,k)枚举，每次找到取值w，算出左右区间。左区间：l=i，右区间：当⌊ki⌋=w时，根据高斯消元的性质，w=⌊ki⌋≤ki

so,i≥kw,r=kw

由于可能存在r≥n，r=min(r，w)

对于每个区间，ans+=w∗∑q=lrq，然后到下一区间：i=r+1

结果为ans

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
long long n,m,ans;
inline long long js(long long x)
{
    return x*(x+1)>>1;
}
int main()
{
    long long i,j;
    scanf("%I64d%I64d",&n,&m);
    for (i=1,j=0;i<=min(n,m);i=j+1)
     {
        j=m/(m/i);//计算一段数的结尾 
        if (j>n) j=n;//如果j跑到n之外，就把它拽回n 
        ans+=m/i*((js(j)-js(i-1)));//用前缀和来维护每一段的和 
     }
    ans=n*m-ans; 
    printf("%I64d\n",ans); 
    return 0;
 }