1. 程式人生 > >[BZOJ1257]余數之和

[BZOJ1257]余數之和

input span ret urn space efi 一行 main ace

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。 1<=n ,k<=10^9

Output

輸出僅一行,即j(n, k)。

Sample Input

5 3

Sample Output

7

HINT

Source

把$a$%$b$表示成$a - \lfloor\frac{a}{b}\rfloor \cdot b$的形式

然後整除分塊搞一波就行了

代碼:

 1 #include<iostream>
 2 #include<cstdio>
 3 #define ll long long
 4 using namespace std;
 5 ll k,n,ans;
 6 int main()
 7 {
 8     cin>>n>>k;
 9     if(n>k) 
10     {
11         ans=(n-k)*k;
12 n=k; 13 } 14 for(ll r,l=1;l<=n;l=r+1) 15 { 16 r=k/(k/l); 17 if(r>n) r=n; 18 ans+=(r-l+1)*k-(r-l+1)*(l+r)/2*(k/l); 19 } 20 cout<<ans; 21 return 0; 22 }

[BZOJ1257]余數之和