258. 各位相加 - H1S97X's Blog

Problem#

给定一个非负整数 num，反复将各个位上的数字相加，直到结果为一位数。返回这个结果。

https://leetcode.cn/problems/add-digits/description/

示例 1：

输入：num = 38 输出：2 解释：3 + 8 = 11, 1 + 1 = 2。

示例 2：

输入：num = 0 输出：0

约束：

0 <= num <= 2³¹ - 1

思路#

解法 1：模拟#

循环取各位数字之和，直到结果为一位数。

解法 2：数学公式（数根）#

数根（digital root）公式：1 + (num - 1) % 9，当 num 为 0 时结果为 0。

1
class Solution:
2
    def addDigits(self, num: int) -> int:
3
        if num == 0:
4
            return 0
5
        return 1 + (num - 1) % 9

1
import unittest
2

3

4
class TestAddDigits(unittest.TestCase):
5
    def setUp(self):
6
        self.solution = Solution()
7

8
    def test_example1(self):
9
        self.assertEqual(self.solution.addDigits(38), 2)
10

11
    def test_zero(self):
12
        self.assertEqual(self.solution.addDigits(0), 0)
13

14
    def test_single(self):
15
        self.assertEqual(self.solution.addDigits(5), 5)
16

17
    def test_large(self):
18
        self.assertEqual(self.solution.addDigits(999), 9)
19

20

21
if __name__ == '__main__':
22
    unittest.main()