题目传送门:https://www.acwing.com/problem/content/201/
[题目大意]:
给出正整数n和k,计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值。
例如j(5, 3)=3 mod 1 + 3 mod 2 + 3 mod 3 + 3 mod 4 + 3 mod 5=0+1+0+3+3=7。
[输入格式]:输入仅一行,包含两个整数n, k。
[输出格式]:输出仅一行,即j(n, k)。
[数据范围]:1≤n,k≤10^9
[输入样例]:5 3
[输出样例]:7
分析:k mod i = k - floor(k / i )* i,可将j(n,k)转化成如下式子:
f(n,k) = k mod 1+ k mod 2 + …k mod n
= k - floor(k/1) * 1 + k - floor(k/2) * 2+… k - floor(k /n) *n
= n * k - sum(floor(k/i) * i) 其中 i = 1~ n 。
接下来我们拿n= 9 ,k = 4 为例分析下k/i:
4/1 4/2 4/3 4/4 4/5 4/6 4/7 4/8 4/9
4 2 1 1 0 0 0 0 0
后面的sum(floor(k/i) * i) = 4 * 1 + 2 * 2 + 1 *(3+ 4)+ 0 *(5+6+7+8+9),括号里的都是等差数列,可以由高斯求和的方法处理。那么我们就需要知道哪几个连续的商是相同的呢?那么在当前的i数时,找到与k/i想等的最大的gx,使得k / gx = k/ i. 因此gx= k / floor(k/i),为了避免算出来的gx超出n,因此准确来说gx = min(k / floor(k/i) , n);这样便可以将i~ gx之间的数用高斯求和处理了。然后继续下次从gx+ 1为起点寻找连续块。
详细代码如下:
#include<bits/stdc++.h>
using namespace std;
int main(){
long long n,k,ans;
cin >> n >> k;
ans = n * k;
int gx;
for(int i = 1;i<= n; i= gx + 1 ){
gx = k/i!=0 ?min( k / (k / i), n ) : n; // 找到与k/i相等的 k/x中最大的x,因为k/i ~ k/x相等的商,使得这段里是等差序列。
ans = ans - k/i * (i + gx)* (gx - i + 1) / 2;
}
cout << ans << endl;
return 0;
}
/*
没有写min(k / (k /i) , n) 导致错误的测试数据是: 9 53
当i == 9时,gx 计算出来是10,但是,10 已不在要求的数据范围内,因此gx需要为 9 才正确。
因此gx = k / x ? min(k/(k/i),n) :n 此处min一定要写。
*/