[BZOJ1257]余數之和
阿新 • • 發佈:2018-09-01
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=7Input
輸入僅一行,包含兩個整數n, k。 1<=n ,k<=10^9
Output
輸出僅一行,即j(n, k)。
Sample Input
5 3Sample Output
7HINT
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]余數之和