题目描述
给定由若干 0 和 1 组成的数组 A。我们定义 N_i:从 A[0] 到 A[i] 的第 i 个子数组被解释为一个二进制数(从最高有效位到最低有效位)。
返回布尔值列表 answer,只有当 N_i 可以被 5 整除时,答案 answer[i] 为 true,否则为 false。
示例
输入:[0,1,1]
输出:[true,false,false]
解释:
输入数字为 0, 01, 011;也就是十进制中的 0, 1, 3 。只有第一个数可以被 5 整除,因此 answer[0] 为真。
输入:[1,1,1]
输出:[false,false,false]
输入:[0,1,1,1,1,1]
输出:[true,false,false,false,true,false]
输入:[1,1,1,0,1]
输出:[false,false,false,false,false]
解答
重点在于二进制数的计算
class Solution(object):
def prefixesDivBy5(self, A):
"""
:type A: List[int]
:rtype: List[bool]
"""
ans = list()
pre = 0
for bit in A:
# 上一次计算结果*2 + 当前二进制数最后一位 = 当前二进制数的十进制表示
cur = pre*2 + bit
pre = cur
if cur%5==0:
ans.append(True)
else:
ans.append(False)
return ans