# 给你一个非负整数 num ,请你返回将它变成 0 所需要的步数。 如果当前数字是偶数,你需要把它除以 2 ;否则,减去 1 。
#
#
#
# 示例 1:
#
# 输入:num = 14
# 输出:6
# 解释:
# 步骤 1) 14 是偶数,除以 2 得到 7 。
# 步骤 2) 7 是奇数,减 1 得到 6 。
# 步骤 3) 6 是偶数,除以 2 得到 3 。
# 步骤 4) 3 是奇数,减 1 得到 2 。
# 步骤 5) 2 是偶数,除以 2 得到 1 。
# 步骤 6) 1 是奇数,减 1 得到 0 。
#
#
# 示例 2:
#
# 输入:num = 8
# 输出:4
# 解释:
# 步骤 1) 8 是偶数,除以 2 得到 4 。
# 步骤 2) 4 是偶数,除以 2 得到 2 。
# 步骤 3) 2 是偶数,除以 2 得到 1 。
# 步骤 4) 1 是奇数,减 1 得到 0 。
#
#
# 示例 3:
#
# 输入:num = 123
# 输出:12
#
#
#
#
# 提示:
#
#
# 0 <= num <= 10^6
#
# Related Topics 位运算 数学 👍 72 👎 0
初级做法:
- 用 x & 1 == 1 判断x是否是奇数
- 用 x >> 1 来实现除以2的操作
# leetcode submit region begin(Prohibit modification and deletion)
class Solution:
def numberOfSteps(self, num: int) -> int:
ret = 0
while num != 0:
num = num - 1 if num & 1 else num >> 1
ret += 1
return ret
# leetcode submit region end(Prohibit modification and deletion)
减少迭代次数
- 奇数时候,要减1再除2;
- 偶数时候,直接除2;
都要除2,只不过奇数的情况多消耗一步。但最后一步除外,当最后一位是1时,减1直接为0了,无需再除2。
那么可以统一这两种情况。
class Solution:
def numberOfSteps(self, num: int) -> int:
ret = 0
while num:
ret += num & 1
if num > 1:
ret += 1
num >>= 1
return ret
直接计算
本质上:
- 每个二进制位的1都会带来两步的代价
- 每个二进制位的0都会带来一步的代价
只需计算出1和0的个数即可。
class Solution:
def numberOfSteps(self, num: int) -> int:
return len(b:=bin(num)[2:]) + b.count('1') - 1
作者:himymBen
链接:https://leetcode-cn.com/problems/number-of-steps-to-reduce-a-number-to-zero/solution/pythonjavajavascriptgo-tou-guo-xian-xian-2ser/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
其中:
bin(num)
将数字转为二进制:
bin(14)
>> '0b1110'
( b:= x )
是赋值语句,在这只是为了缩短代码行数。
除夕快乐!