网易招聘笔试题--数对

题目链接： https://www.nowcoder.com/question/next?pid=9763997&qid=152611&tid=14751374

解题思路： 考虑固定每个y，那么对于y=k+1,,..., n,生成的符合条件的数对共有 n / y * (y - k) + (n % y - k + 1)  |  (n % y > 0)。

                  对所有的y求和即可。特例k =0 时， ans = n * n;

#include <bits/stdc++.h>
using namespace std;

long long n, k;

int main()
{
    long long ans = 0;
    scanf("%lld%lld", &n, &k);
    if (k == 0LL)
    {
        printf("%lld
", n * n);
        return 0;
    }
    for (long long y = k + 1; y <= n; ++y)
    {
        ans += n / y * (y - k);
        if (n % y >= k)
            ans += (n % y - k + 1);
    }
    printf("%lld
", ans);
    return 0;
}