bzoj 1257

1257: [CQOI2007]餘數之和sum

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

代碼：

//k%i==k-[k/i]*i;而後就能夠分段求了。每一段求等差數列和再乘[k/i]。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int main()
{
    ll n,k;
    scanf("%lld%lld",&n,&k);
    ll ans=0;
    if(n>k) { ans=k*(n-k);n=k; }
    ll i=1;
    while(i<=n){
        ll p=k/i,last=k/p;
        if(last>n) last=n;
        ans+=(last-i+1)*k-p*(i+last)*(last-i+1)/2;
        i=last+1;
    }
    printf("%lld\n",ans);
    return 0;
}