给你一个整数 n,请你判断该整数是否是 2 的幂次方。如果是,返回 true ;否则,返回 false 。如果存在一个整数 x 使得 n == 2x ,则认为 n 是 2 的幂次方。
示例 1:
输入:n = 1
输出:true
解释:20 = 1
示例 2:
输入:n = 16
输出:true
解释:24 = 16
示例 3:
输入:n = 3
输出:false
示例 4:
输入:n = 4
输出:true
示例 5:
输入:n = 5
输出:false
思路与算法
一个数n是 2的幂,当且仅当 n是正整数,并且 n的二进制表示中仅包含 1个 1。因此我们可以考虑使用位运算,将 n的二进制表示中最低位的那个 1提取出来,再判断剩余的数值是否为0即可。下面介绍两种常见的与「二进制表示中最低位」相关的位运算技巧。
第一个技巧是 n & (n-1)
该位运算技巧可以直接将二进制表示的最低位 1 移除,它的原理如下:
假设n 的二进制表示为 a 10000,其中 a 表示若干个高位, n-1的二进制表示为a 011111。我们将n与n-1进行按位与运算,高位a不变,在这之后的所有低位都会变为0,这样我们就将最低位的那个 1 移除了。因此,如果 n是正整数并且 n & (n-1) = 0 ,那么 n 就是 2 的幂。
第二个技巧是 n&(-n)
该位运算技巧可以直接获取二进制表示的最低位的 1。
由于负数是按照补码规则在计算机中存储的, -n的二进制表示为 a^100000,即每一位取反再加上 1. 按位与运算后,高位全部变为 0,低位不变,这样我们就获取了二进制表示的最低位的1。因此,如果n是正整数并且n&(-n)=n ,那么 n 就是 2 的幂。
时间,空间复杂度都是O(1)
class Solution:
def isPowerOfTwo(self, n: int) -> bool:
if (n > 0) and (n & (n-1) == 0):
return True
else:
return False