Easy
Binary Prefix Divisible By 5 — C++
Full explanation · Time O(n) · Space O(1)
// Time: O(n)
// Space: O(1)
class Solution {
public:
vector<bool> prefixesDivBy5(vector<int>& A) {
for (int i = 1; i < A.size(); ++i) {
A[i] += A[i - 1] * 2 % 5;
}
vector<bool> result(A.size());
transform(A.cbegin(), A.cend(), result.begin(),
[](int x) { return x % 5 == 0; });
return result;
}
};