函数练习题——腾讯笔试编程题

题目需求：对于一个十进制的正整数，定义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)  # 1.先把数字转换为字符串
    sum = 0
    for i in n:
        sum += int(i) ** 2
    return sum      # 2.计算字符串中每个数的平方
s = raw_input('请输入(k,a,b)：')    # 1.接收变量
li = []
for item in s.split():
    li.append(int(item))
k, a, b = li
count = 0   # 2.进行判断是否满足条件
for i in range(a, b + 1):
    if k * f(i) == i:
        count += 1
print count