网易2019实习生招聘-数对

[编程题] 数对

时间限制：1秒

空间限制：32768K

牛牛以前在老师那里得到了一个正整数数对(x, y), 牛牛忘记他们具体是多少了。

但是牛牛记得老师告诉过他x和y均不大于n, 并且x除以y的余数大于等于k。

牛牛希望你能帮他计算一共有多少个可能的数对。

输入描述:

输入包括两个正整数n,k(1 <= n <= 10^5, 0 <= k <= n - 1)。

输出描述:

对于每个测试用例, 输出一个正整数表示可能的数对数量。

输入例子1:

5 2

输出例子1:

7

例子说明1:

满足条件的数对有(2,3),(2,4),(2,5),(3,4),(3,5),(4,5),(5,3)

using namespace std;

#include<iostream>
long long add(long long a){
if (a == 1){
return 1;
}
return a + add(a - 1);
}
int main(){
int n, k;
cin >> n >> k;
   long long sum = 0; //先写当K<=x<y的情况，此时对于任意x有n-y个y符合
   sum = sum + add(n - k); 

//接下来写 K<Y<x的情况
for (int i = k + 1; i <n; i++){
if (n%i == i-1){
sum = sum+(n / i)*(i - k);
// cout <<"y1"<< " "<<i << endl;
}
else{
sum = sum + ((n/ i) - 1)*(i - k);
//cout << "y2" <<" "<<i << endl;
if ((n%i) - k >= 0){
sum = sum+((n%i) - k )+ 1;
// cout << "y3" <<" "<< i << endl;
}
}
}
if (k == 0){
sum = sum - n+1;
}


cout << sum;
return 0;
}