python定义功能为各个位平方和的函数，给出整数k,a,b 计算整数n，满足n属于a,b之间，并且k 乘 f(n)等于n

代码：

def f(n):
    n = str(n) #先把数字转化为字符串
    sum = 0
    for i in n:     
        sum += int(i) ** 2 #计算字符串中每个数的平方
    return sum

s = raw_input('请输入(k,a,b):')
li = []
for item in s.split():      #将输入的字符串以空格划分，依次遍历，添加到li列表中
    li.append(int(item))
k, a, b = li   #赋值给k,a,b

count = 0
for i in range(a, b + 1):  #条件1满足a<=n<=b
    if k * f(i) == i:   #条件2满足k*f(n)=n
        count += 1  
print count  #输出满足条件的总个数