求小于等于n的数中,其二进制中不含相邻两个1的数的个数。
解题思路:
首先考虑这样一个问题,对于长度小于等于n的二进制数中,满足上述条件的数有多少个。这个问题的解决方法如下
# a[i]表示以0结尾长度为i的数的个数
# b[i]表示以1结尾长度为i的数的个数
a = [1]*n
b = [1]*n
我们有如下递推式
for i in range(1,n):
a[i] = a[i-1] + b[i-1]
b[i] = a[i-1]
res = a[-1] + b[-1]
对于a[i],他的上一位既可以是0也可以是1,即以0或1结尾长度为i-1的数的个数,所以有
a[i] = a[i-1] + b[i-1]
对于b[i],他的上一位只能是0,即以1结尾的长度为i-1的数的个数,所以有
b[i] = a[i-1]
这样我们就能求得长度为n的二进制满足条件的个数。
考虑 nuber = 9(二进制为1001),对应的 n= 4,我们可以轻松的求n=4时:
a = [1,2,3,5]
b = [1,1,2,3]
res = 8
但 n = 4的时候对应的十进制数是15,我们要求的是小于等于10的满足条件的数,怎么办呢,我们从左往右遍历。
每次找到有相邻的两个0(index 分别为i和i+1),那么我们就将res 减去b[length-i-2](length-i-1对应的是i+1到末尾的字符串的长度),因为res是第一个0后面既可以跟1又可以跟0的情况下求的结果,现在后面只有0没有1,所以我们要把对应的1产生的满足条件的数给减掉。
如果遇到两个1就直接退出,以为不会出现两个1的情况,出现两个1肯定后面覆盖了10开头的所有情况。
AC代码
class Solution(object):
def findIntegers(self, num):
"""
:type num: int
:rtype: int
"""
num = bin(num)[2:]
length = len(num)
a = [1]*length
b = [1]*length
for i in range(1,length):
a[i] = a[i-1] + b[i-1]
b[i] = a[i-1]
result = a[-1] + b[-1]
for i in range(length-1):
if num[i] == "0" and num[i+1] == "0":
result -= b[length - i -2]
if num[i] == "1" and num[i+1] == "1":
break
return result