题目地址:
https://leetcode.com/problems/binary-prefix-divisible-by-5/
给定一个数组,要求返回一个boolean数组,使得等于组成的二进制表示的数是否能被整除。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
public List<Boolean> prefixesDivBy5(int[] A) {
List<Boolean> res = new ArrayList<>(A.length);
int cur = 0;
for (int i = 0; i < A.length; i++) {
// 算出前i个数字二进制表示的数后直接模5后判断
cur = ((cur << 1) + A[i]) % 5;
res.add(cur == 0);
}
return res;
}
}
时间复杂度,空间。
算法正确性证明:
对于,求对的带余除法,其中。我们证明对于任意,有。证明如下:最后一行显然成立。这就证明了为什么算法中可以直接每一步都取模,而不是保留原数。