可被 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<=A.length<=30000$
$A[i]$ 为 $0$ 或 $1$

令 $N_i$为数组 $A$ 从下标 $0$ 到下标 $i$ 的前缀表示的二进制数，则有 $N_0=A[0]$当 $i>0$ 时，有 $N_i=N_{i-1}\times 2+A[i]$。令 $n$ 为数组 $A$ 的长度，则对于 $0\le i<n0\le i<n$，可以依次计算每个 $N_i$的值。对于每个 $N_i$的值，判断其是否可以被 $5$ 整除，即可得到答案。

考虑到数组 $A$ 可能很长，如果每次都保留 $N_i$的值，则可能导致溢出。由于只需要知道每个 $N_i$是否可以被 $5$ 整除，因此在计算过程中只需要保留余数即可。

令 $M_i$表示计算到下标 $i$ 时的除以 $5$ 的余数，则有 $M_0=A[0]$（显然 $A[0]$ 一定小于 $5$），当 $i>0$ 时，有 $M_i=(M_{i-1}\times 2+A[i])\mathrm{m}\mathrm{o}\mathrm{d}5$，判断每个 $M_i$是否为 $0$ 即可。由于 $M_i$一定小于 $5$，因此不会溢出。

如何证明判断 $M_i$是否为 $0$ 可以得到正确的结果？可以通过数学归纳法证明。

当 $i=0$时，由于 $N_0=A[0]<5，因此 $M_0=N_0$,$M_0=N_0\mathrm{m}\mathrm{o}\mathrm{d}5$成立。

当 $i>0$ 时，假设 $M_{i-1}=N_{i-1}\mathrm{m}\mathrm{o}\mathrm{d}5$成立，考虑 $N_i\mathrm{m}\mathrm{o}\mathrm{d}5$和 $M_i$
的值：
$$\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+A[i]\mathrm{m}\mathrm{o}\mathrm{d}5\\ & \\ M_i=& (M_{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+A[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+A[i]\mathrm{m}\mathrm{o}\mathrm{d}5\\ =& (N_{i-1}\times 2)\mathrm{m}\mathrm{o}\mathrm{d}5+A[i]\mathrm{m}\mathrm{o}\mathrm{d}5\end{array}$$

即可得到正确的结果。

class Solution {
public:
    vector<bool> prefixesDivBy5(vector<int>& A) {
        int n=A.size();
        vector<bool>res;
        int num=0;
        for (int i=0;i<n;i++){
            num=(num*2+A[i])%5;
            if (num==0){
                res.push_back(true);
            }else res.push_back(false);
        }
        return res;
    }
};