LeetCode 1018. 可被 5 整除的二进制前缀 | Python

1018. 可被 5 整除的二进制前缀

---

题目来源：力扣（LeetCode）https://leetcode-cn.com/problems/binary-prefix-divisible-by-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] 为真。

示例 2：

输入：[1,1,1]
输出：[false,false,false]

示例 3：

输入：[0,1,1,1,1,1]
输出：[true,false,false,false,true,false]

示例 4：

输入：[1,1,1,0,1]
输出：[false,false,false,false,false]

提示：

1. 1 <= A.length <= 30000
2. A[i] 为 0 或 1

解题思路

---

思路：模拟

先审题，题目给定一个数组 $A$，其中元素由若干个 $0$ 或 $1$ 组成。

题目定义 $N_i$：表示从 $A[0]$ 到 $A[i]$ 的第 $i$ 个子数组被解释为一个二进制。（由高位到低位）。

题目要求返回一个布尔值列表 $\mathrm{a}\mathrm{n}\mathrm{s}\mathrm{w}\mathrm{e}\mathrm{r}$，其中值由 $N_i$ 是否能被 $5$ 整除来决定，若能被整除，值为 $True$，否则为 $False$。

这里再说下关于 $N_i$ 的定义，结合示例 $1$ 来看：

输入：[0,1,1]
输出：[true,false,false]

• 在这里 $N_0$ 为 $0$，对应十进制数 $0$；

• $N_1$ 为 $01$，这里表示由数组 $A$ 索引 $0$ 到 $1$ 两个元素组成的二进制数，对应十进制数 $1$；

• 那么 $N_2$ 在这里为 $011$，对应十进制数 $3$。

由上面的例子，我们也可以看到：

• 当 $i=0$ 时，$N_0=A[0]$ ；
• 当 $i>0$ 时，$N_i=N_{i-1}\times 2+A[i]$ 。

也就是说依次计算每个 $N_i$ 的值，判断是否被 $5$ 整除即可。

但由于题目后面提示数组 $A$ 的长度 $1\le A.length\le 30000$，这里如果每次都需要计算保存 $N_i$ 的值，可能会导致溢出。

因为我们只需要知道 $N_i$ 是否能够被 $5$ 整除，那么在计算的过程中考虑只保留余数。因为在这里余数只可能是 $[0,1,2,3,4]$ 之中的其中一个，不用担心溢出的问题。

现在用 $R_i$ 表示计算下标 $i$ 时除以 $5$ 的余数。因为数组 $A$ 的元素只有 $0$ 或 $1$，那么有 $R_0=A[0]$，当 $i>0$ 时，则有 $R_i=(R_{i-1}\times 2+A[i])\mathrm{m}\mathrm{o}\mathrm{d}5$，那么在这里只需要判断每个 $R_i$ 是否为 $0$ 即可。

现在主要要证明这个仅判断余数是否为 $0$，能否成立？这里用数学归纳法来证明。

当 $i=0$ 时，由于 $N_0=A[0]$，而 $R_0=A[0]$，可得 $R_0=N_0$ 且 $A[0]$ 仅可能为 $0$ 或 $1$，那么 $R_0=N_0\mathrm{m}\mathrm{o}\mathrm{d}5$ 成立；

当 $i>0$ 时，现在假设 $R_{i-1}=N_{i-1}\mathrm{m}\mathrm{o}\mathrm{d}5$ 成立，证明 $R_i=N_i\mathrm{m}\mathrm{o}\mathrm{d}5$ 成立：
$$\begin{array}{rl}{N}_i\mathrm{m}\mathrm{o}\mathrm{d}5& =(N_{i-1}\times 2+A[i])\mathrm{m}\mathrm{o}\mathrm{d}5\\ & =(N_{i-1}\times 2)\mathrm{m}\mathrm{o}\mathrm{d}5+\mathrm{A}[\mathrm{i}]\mathrm{m}\mathrm{o}\mathrm{d}5\\ & \\ R_i& =(R_{i-1}\times 2+A[i])\mathrm{m}\mathrm{o}\mathrm{d}5\\ & =(N_{i-1}\mathrm{m}\mathrm{o}\mathrm{d}5\times 2+\mathrm{A}[\mathrm{i}])\mathrm{m}\mathrm{o}\mathrm{d}5\\ & =(N_{i-1}\mathrm{m}\mathrm{o}\mathrm{d}5\times 2)\mathrm{m}\mathrm{o}\mathrm{d}5+\mathrm{A}[\mathrm{i}]\mathrm{m}\mathrm{o}\mathrm{d}5\\ & =(N_{i-1}\times 2)\mathrm{m}\mathrm{o}\mathrm{d}5+\mathrm{A}[\mathrm{i}]\mathrm{m}\mathrm{o}\mathrm{d}5\end{array}$$
由此可得，$R_i=N_i\mathrm{m}\mathrm{o}\mathrm{d}5$ 成立。

将上面的情况结合起来，也就说当 $0\le i<n$ 时，$R_i=N_i\mathrm{m}\mathrm{o}\mathrm{d}5$ 恒成立。那么只要计算$R_i$ 的值判断是否为 0 即可。

具体的代码实现如下。

class Solution:
    def prefixesDivBy5(self, A: List[int]) -> List[bool]:
        answer = []
        rem = 0
        for num in A:
            rem = ((rem << 1) + num) % 5
            answer.append(rem == 0)
        
        return answer

欢迎关注

---

公众号 【书所集录】

---

如有错误，烦请指出，欢迎指点交流。若觉得写得还不错，麻烦点个赞👍，谢谢。