请问谁有好的方法求各位数的平方和，并且不确定是几位数（10的18次方以内）

Question

3、5. 对于一个十进制的正整数， 定义f(n)为其各位数字的平方和，如:
f(13) = 12 + 32 = 10
f(207) = 22 + 02 + 72 = 53
下面给出三个正整数k，a, b,你需要计算有多少个正整数n满足a<=n<=b, 且k*f(n)=n
输入:第一行包含3个正整数k，a, b, k>=1, a,b<=1018, a<=b;
输出：输出对应的答案;

范例:输入: 51 5000 10000
输出: 3

Answer 1

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


def main(k, a, b):
    result = 0
    for i in range(a, b):
        if k * f(i) == i:
            result += 1
    return result

if __name__ == '__main__':
    print(main(51, 5000, 10000))