函数练习--腾讯笔试题

"""
腾讯笔试编程题：
题目需求:
    对于一个十进制的正整数， 定义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：
def f(n):
    #1.先把数字转换为字符串
    n = str(n)
    #2.计算字符串中每个数的平方
    sum = 0
    for item in n:
        sum += int(item) ** 2
    return sum

# print(f(207))

#1.接收变量 k,a,b
s = input()
#2.存储整型k,a,b
li = []
for item in s.split():
    li.append(int(item))

k,a,b = li

#3.进行判断是否满足条件
count = 0
for i in range(a,b+1):
    if k*f(i) == i:
        count += 1
print(count)

方法2
def fun(k):
    n = str(k)         #先把数字转换成字符串
    sum = 0
    for i in n:
        sum += int(i)**2   #计算字符串中每个数的平方
    return sum
s = input('请输入(k,a,b):')   #接受变量
li = []
for i in s.split():
    li.append(int(i))
k,a,b = li
count = 0                #判断是否满足条件
for i in range(a,b+1):
    if k * fun(i) == i:
        count+=1
print(count)