python——腾讯笔试编程题（函数）

腾讯笔试编程题：

题目需求:

• 对于一个十进制的正整数， 定义f(n)为其各位数字的平方和，如:
  f(13) = 12 + 32 = 10
  f(207) = 22 + 02 + 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

#1.定义函数f(n):用于实现计算各位数的平方和
def f(n):
    s = str(n)  ##先将接收的数字转换成字符串类型，方便遍历
    sum = 0
    for i in s:   ##便历数字的每一位
        sum += int(i)**2    ##计算各位数的平方和
    return sum

#2.存储整形k，a，b
s = input()  ##题中没有要求显示任何东西，直接input即可
li = []
for item in s.split():  ##便历接收的字符串
    li.append(int(item))  ##将字符串转换成int型，添加到列表li中
k,a,b = li     ##存储整形

'''
分开写：
s = input()
spilt = s.split(' ')
k = int(spilt[0])
a = int(spilt[1])
b = int(spilt[2])

'''
#3.进行判断是否满足条件
count = 0   ##定义一个计数器
for n in range(a,b+1):
    if k * f(n) == n:
        count += 1
print(count)

结果：
51 5000 10000
3