笔试:牛牛数对问题 python

# encoding=utf-8
'''
牛牛以前在老师那里得到了一个正整数数对(x, y), 牛牛忘记他们具体是多少了。
但是牛牛记得老师告诉过他xy均不大于n, 并且x除以y的余数大于等于k牛牛希望你能帮他计算一共有多少个可能的数对。
输入包括两个正整数n,k(1 <= n <= 10^5, 0 <= k <= n - 1)输出描述:

对于每个测试用例, 输出一个正整数表示可能的数对数量。
示例1
输入
5 2
输出
7
说明
满足条件的数对有(2,3),(2,4),(2,5),(3,4),(3,5),(4,5),(5,3)
'''
if __name__ == '__main__':
    str = raw_input()
    ls = [int(n) for n in str.split()]
    n = ls[0]
    k = ls[1]
    res = 0

    for y in range(max(k, 1), n + 1):
        res += n/y*(y - k)
        if n % y >= k:
            if k:
                res += n % y - k + 1
            else:
                res += n % y
    print res

你可能感兴趣的:(python编程)