bzoj 1257 余数之和 数论

1257: [CQOI2007]余数之和sum

Time Limit: 5 Sec Memory Limit: 162 MB
Submit: 4837 Solved: 2249
[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

#include <cstdio>
#include <iostream>
#include <cstring>
#define ll long long 
using namespace std;
int n,k;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int main(){
    n=read();k=read();
    ll ans=0;
    if(n>k){
        ans=(ll)(n-k)*k;
        n=k;
    }
    int r;
    for(int i=1;i<=n;i=r+1){
        int t=k/i; r=k/t;
        if(r>=n) r=n;
        ans+=(ll)(r-i+1)*k-(ll)(r-i+1)*(i+r)/2*t;
    }
    printf("%lld",ans);
    return 0;
}