326. 3 的幂 - H1S97X's Blog

Problem#

给定一个整数，写一个函数来判断它是否是 3 的幂次方。如果是，返回 true；否则，返回 false。

如果存在一个整数 x 使得 n == 3ˣ，则认为 n 是 3 的幂次方。

https://leetcode.cn/problems/power-of-three/description/

示例 1：

输入：n = 27 输出：true

示例 2：

输入：n = 0 输出：false

示例 3：

输入：n = 9 输出：true

约束：

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

思路#

循环除以 3，直到不能整除为止，最后判断是否等于 1。

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

1
import unittest
2

3

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

8
    def test_twentyseven(self):
9
        self.assertTrue(self.solution.isPowerOfThree(27))
10

11
    def test_nine(self):
12
        self.assertTrue(self.solution.isPowerOfThree(9))
13

14
    def test_zero(self):
15
        self.assertFalse(self.solution.isPowerOfThree(0))
16

17
    def test_not_power(self):
18
        self.assertFalse(self.solution.isPowerOfThree(45))
19

20
    def test_negative(self):
21
        self.assertFalse(self.solution.isPowerOfThree(-3))
22

23

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