263. 丑数 - H1S97X's Blog

Problem#

丑数就是只包含质因数 2、3 和 5 的正整数。

给你一个整数 n，请你判断 n 是否为丑数。如果是，返回 true；否则，返回 false。

https://leetcode.cn/problems/ugly-number/description/

示例 1：

输入：n = 6 输出：true 解释：6 = 2 × 3

示例 2：

输入：n = 1 输出：true 解释：1 通常被视为丑数。

示例 3：

输入：n = 14 输出：false 解释：14 不是丑数，因为它包含了质因数 7。

约束：

-2³¹ <= n <= 2³¹ - 1

思路#

反复除以 2、3、5，直到不能整除为止。最后判断剩余是否为 1。注意 n <= 0 不是丑数。

1
class Solution:
2
    def isUgly(self, n: int) -> bool:
3
        if n <= 0:
4
            return False
5
        for p in (2, 3, 5):
6
            while n % p == 0:
7
                n //= p
8
        return n == 1

1
import unittest
2

3

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

8
    def test_six(self):
9
        self.assertTrue(self.solution.isUgly(6))
10

11
    def test_one(self):
12
        self.assertTrue(self.solution.isUgly(1))
13

14
    def test_fourteen(self):
15
        self.assertFalse(self.solution.isUgly(14))
16

17
    def test_zero(self):
18
        self.assertFalse(self.solution.isUgly(0))
19

20
    def test_negative(self):
21
        self.assertFalse(self.solution.isUgly(-6))
22

23

24
if __name__ == '__main__':
25
    unittest.main()