萌新练习写代码的每日一练:可被 5 整除的二进制前缀

给定由若干 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题

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值