给定由若干 0 和 1 组成的数组 A。我们定义 N_i:从 A[0] 到 A[i] 的第 i 个子数组被解释为一个二进制数(从最高有效位到最低有效位)。
返回布尔值列表 answer,只有当 N_i 可以被 5 整除时,答案 answer[i] 为 true,否则为 false。
示例 1:
输入:[0,1,1]
输出:[true,false,false]
解释:
输入数字为 0, 01, 011;也就是十进制中的 0, 1, 3 。只有第一个数可以被 5 整除,因此 answer[0] 为真。
思路:运用我学长的一句圣经“万物皆可数学”,先定义一个数now = 0,之后对now进行左移操作并加上当前列表中的值,然后对5进行取余操作判断即可。
代码:
class Solution:
def prefixesDivBy5(self, A: List[int]) -> List[bool]:
res = []
now = 0
for i in A:
now = (now << 1) + i
res.append(not now % 5)
return res
但是这样会出现一个问题,假如列表A的长度过大,比如len(A) = 30000,在这么操作的话会出现溢出的情况,ll都救不回来的那种(误
因此需要进行优化,这时候利用一个公式
m≡n(mod 5) 则ma≡na(mod 5)
就可以进行分布操作,每次移位操作之后直接对5取余,是和最终结果对5取余的效果是一样的,而且每次mod5之后的结果只可能是0、1、2、3、4,利用上一步的结果做后续运算,省时省力,不会溢出。
代码如下:
class Solution:
def prefixesDivBy5(self, A: List[int]) -> List[bool]:
res = []
now = 0
for i in A:
now = ((now << 1) + i) % 5
res.append(not now)
return res
不得不说数学真好玩(考试的时候除外
题目来自LeetCode第1018题