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

目录结构

1.题目

2.题解

---

1.题目

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-prefix-divisible-by-5
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

主要注意溢出问题，可每次左移一位后对5取余。

public class Solution1018 {

    @Test
    public void test1018() {
        int[] A = {0, 1, 1, 1, 1, 1};
        System.out.println(prefixesDivBy5(A));
    }

    public List<Boolean> prefixesDivBy5(int[] A) {
        List<Boolean> result = new ArrayList<>();
        int num = 0;
        for (int a : A) {
            num = (num << 1 ) + a;
            if (num % 5 == 0) {
                result.add(true);
            } else {
                result.add(false);
            }
            num %=5;
        }
        return result;
    }
}

• 时间复杂度：
• 空间复杂度：