校招真题练习022 数对(网易)

数对

题目描述
牛牛以前在老师那里得到了一个正整数数对(x, y), 牛牛忘记他们具体是多少了。
但是牛牛记得老师告诉过他x和y均不大于n, 并且x除以y的余数大于等于k。
牛牛希望你能帮他计算一共有多少个可能的数对。

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

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

 1 def main():
 2     n, k = map(int, input().split())
 3     count = 0
 4     if k == 0:
 5         count = n * n
 6     else:
 7         for y in range(k+1, n+1):
 8             count += (n // y) * (y - k)
 9             if n % y >= k:
10                 count += (n % y - k + 1)
11     print(count)
12 
13 if __name__ == '__main__':
14     main()

不会做，一遇到这种数学题目就蒙。

参考：https://www.nowcoder.com/profile/3865993/codeBookDetail?submissionId=25827709