给定n，取n的位数之和。如果该值超过一位，则继续减少一位数

Question

def digital_root(n):
    x = str(n)
    while len(x)!=0 and len(x)!=1:
        r = 0
        for i in range(len(x)):
            r= r + int(x[i])
        x = str(r)
    return r

数字根是一个数字中所有数字的递归和。给定n，取n的位数之和。如果该值有多个数字，则继续以这种方式减少，直到产生一个个位数。这仅适用于自然数。

我的代码：

这段代码有效吗？我得到以下错误：

Traceback (most recent call last):
  File "main.py", line 8, in <module>
    test.assert_equals( digital_root(0), 0 )
  File "/home/codewarrior/solution.py", line 8, in digital_root
    return r
UnboundLocalError: local variable 'r' referenced before assignment

Answer 1

如果字符串x的长度为1（您也可以询问while循环的长度是否大于1），代码将直接跳转到返回r。但是，它可能不知道r指的是什么，因为它只在while循环中分配。

Answer 2

我会用递归来处理这个问题

def root_sum(n):
    x = sum(int(i) for i in str(n))
    if x > 9:
       return root_sum(x)
    return x

这是最简单的解决方案

Answer 3

当您到达返回语句时，r看起来超出了范围。它仅在while语句中本地应用，该语句在返回语句之前结束。

最好只有一个返回语句，并且它在函数的最广泛范围内，并且在末尾。因此，尝试在while循环之外声明r，与返回语句的范围相同。该范围也包括time循环内的所有内容。

def digital_root(n):
    x = str(n)
    r = 0
    while len(x) > 1:
        r = 0
        for i in range(len(x)):
            r = r + int(x[i])
        x = str(r)
    return r