# encoding=utf-8 ''' 牛牛以前在老师那里得到了一个正整数数对(x, y), 牛牛忘记他们具体是多少了。 但是牛牛记得老师告诉过他x和y均不大于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
最新推荐文章于 2022-09-30 21:04:50 发布