网易2019实习生招聘笔试-数对

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

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

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

输入描述:

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

输出描述:

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

枚举每一个除数y，对于除数y来说，余数为0-y-1的长度为y的循环节 因此循环部分有n/y*(y-k)个数对（对于被除数小于除数的情况下，有y-k个数对），非循环部分为n%y-k+1个数对，

最后考虑k=0的情况，很明显是n*n个数对

#include <iostream>
#include <cmath>
#include <queue>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstdlib>
#define LL long long
using namespace std;
LL n,k,s;
int main(){
    LL count=0;
    cin>>n>>k;
    if(k==0){
        count=n*n;
    }
    else{
        for(LL i=k+1;i<=n;i++){
        //count+=n-i;
            s=n/i;
            count+=s*(i-k);
            LL m = n%i;
            if(m>=k){
                count+=m-k+1;
            }
        }
    }

    cout<<count<<endl;
    return 0;
}