Python-腾讯笔试编程题

题目需求:
    对于一个十进制的正整数， 定义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

我的主要思路如下：

1.定义一个函数f(n)，先将输入的正整数转换为字符串形式，然后再求各位数字的平方和

2.输入：将输入的值存储到字符串中，然后以空格为分隔符，分离出a,b,k

3.定义一个全局变量Count，用来计数满足条件的个数

4.a到b循环判断是否满足条件，如果满足条件，Count加1，如果不满足，跳过本次循环，进入下一次循环

def f(n):
    result=0
    for i in str(n):
        result += int(i) ** 2
    return result


s=input("请输入：")
k=int(s.split()[0])
a=int(s.split()[1])
b=int(s.split()[2])

Count = 0
for i in range(a,b+1):
    if k*f(i) == i:
        Count += 1
print(Count)