1281. 整数的各位积和之差

Problem#

给你一个整数 n，请你帮忙计算并返回该整数「各位数字之积」与「各位数字之和」的差。

https://leetcode.cn/problems/subtract-the-product-and-sum-of-digits-of-an-integer/description/

示例 1：

输入：n = 234 输出：15 解释：各位数之积 = 2 * 3 * 4 = 24，各位数之和 = 2 + 3 + 4 = 9，24 - 9 = 15

示例 2：

输入：n = 4421 输出：21 解释：各位数之积 = 4 * 4 * 2 * 1 = 32，各位数之和 = 4 + 4 + 2 + 1 = 11，32 - 11 = 21

约束：

1 <= n <= 10⁵

思路#

循环取模 n % 10 得到最后一位，累加/累乘后 n //= 10 去掉最后一位。

1
class Solution:
2
    def subtractProductAndSum(self, n: int) -> int:
3
        product = 1
4
        total = 0
5
        while n:
6
            digit = n % 10
7
            product *= digit
8
            total += digit
9
            n //= 10
10
        return product - total

1
import unittest
2

3

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

8
    def test_example1(self):
9
        self.assertEqual(self.solution.subtractProductAndSum(234), 15)
10

11
    def test_example2(self):
12
        self.assertEqual(self.solution.subtractProductAndSum(4421), 21)
13

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

17
    def test_all_ones(self):
18
        self.assertEqual(self.solution.subtractProductAndSum(111), -1)
19

20

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