39.函数练习---输出满足条件的正整数个数(腾讯笔试练习)

练习：

题目需求:
对于一个十进制的正整数， 定义f(n)为其各位数字的平方和，如:
f(13) = 1 ** 2 + 3 ** 2 = 10
f(207) = 2 ** 2 + 0 ** 2 + 7 ** 2 = 53
下面给出三个正整数k，a, b,你需要计算有多少个正整数n满足a<=n<=b且k * f(n)=n
输入:第一行包含3个正整数k, a, b, k>=1,a,b<=10 ** 18, a<=b;
输出：输出对应的答案;
范例:
输入: 51 5000 10000
输出: 3

代码如下：

def f(n):
    n = str(n)
    sum = 0
    for item in n:
        sum += int(item) ** 2
    return sum
# print(f(13))
# print(f(207))
#接收变量 k a b
s = input()     #'51 5000 10000'
#存储整形k a b
li = []
for item in s.split():
    li.append(int(item))
k,a,b = li
#判断是否满足条件
count = 0
for i in range(a,b+1):
    if k * f(i) == i:
        count += 1
print(count)

结果如下：