题目来自LeetCode,链接:面试题15. 二进制中1的个数。具体描述为:请实现一个函数,输入一个整数,输出该数二进制表示中 1 的个数。例如,把 9 表示成二进制是 1001,有 2 位是 1。因此,如果输入 9,则该函数输出 2。
示例1:
输入:00000000000000000000000000001011
输出:3
解释:输入的二进制串 00000000000000000000000000001011 中,共有三位为 '1'。
示例2:
输入:00000000000000000000000010000000
输出:1
解释:输入的二进制串 00000000000000000000000010000000 中,共有一位为 '1'。
示例3:
输入:11111111111111111111111111111101
输出:31
解释:输入的二进制串 11111111111111111111111111111101 中,共有 31 位为 '1'。
题目就很简单了,直接利用位移运算,将数字不断右移直到变为0为止(注意JAVA得用无符号右移运算符>>>
),这个过程中如果最末一位为1就在结果上+1(可以直接加上n&1
,如果末位为1则为1,否则为0)。时间复杂度为
O
(
l
o
g
n
)
O(logn)
O(logn),空间复杂度为
O
(
1
)
O(1)
O(1)。
JAVA版代码如下:
public class Solution {
// you need to treat n as an unsigned value
public int hammingWeight(int n) {
int result = 0;
while (n != 0) {
result += n & 1;
n >>>= 1;
}
return result;
}
}
提交结果如下:
Python版代码如下:
class Solution:
def cuttingRope(self, n: int) -> int:
if n <= 3:
return n - 1
result = 1
mod = 1000000007
while n > 4:
result = (result * 3) % mod
n -= 3
result = (result * n) % mod
return result
提交结果如下:
或者可以利用n&(n-1)
会使得n
的各位中最后一个1变为0,而其他位不变的特性,通过计算n=n&*(n-1)
直到n==0
需要的次数得到n
中1的个数。时间复杂度为
O
(
m
)
O(m)
O(m)(m
为n
各位中1的个数),空间复杂度为
O
(
1
)
O(1)
O(1)。
JAVA版代码如下:
public class Solution {
// you need to treat n as an unsigned value
public int hammingWeight(int n) {
int result = 0;
while (n != 0) {
n &= n - 1;
++result;
}
return result;
}
}
提交结果如下:
Python版代码如下:
class Solution:
def hammingWeight(self, n: int) -> int:
result = 0
while n != 0:
n &= n - 1
result += 1
return result
提交结果如下: