Given an array A
of 0
s and 1
s, consider N_i
: the i-th subarray from A[0]
to A[i]
interpreted as a binary number (from most-significant-bit to least-significant-bit.)
Return a list of booleans answer
, where answer[i]
is true
if and only if N_i
is divisible by 5.
Example 1:
Input: [0,1,1] Output: [true,false,false] Explanation: The input numbers in binary are 0, 01, 011; which are 0, 1, and 3 in base-10. Only the first number is divisible by 5, so answer[0] is true.
Example 2:
Input: [1,1,1] Output: [false,false,false]
Example 3:
Input: [0,1,1,1,1,1] Output: [true,false,false,false,true,false]
Example 4:
Input: [1,1,1,0,1] Output: [false,false,false,false,false]
Note:
1 <= A.length <= 30000
A[i]
is0
or1
Idea 1. Modular arithmetic
(a*b + c)%d = (a%d) * (b%d) + c%d
Time complexity: O(n)
Space complexity: O(n)
1 class Solution { 2 public List<Boolean> prefixesDivBy5(int[] A) { 3 int val = 0; 4 List<Boolean> divisible = new ArrayList<>(); 5 6 for(int num: A) { 7 val = ((val * 2)%5 + num)%5; 8 divisible.add(val == 0? true : false); 9 } 10 11 return divisible; 12 } 13 }
Idea 1.b left shift, bitwise operation
1 class Solution { 2 public List<Boolean> prefixesDivBy5(int[] A) { 3 int val = 0; 4 List<Boolean> divisible = new ArrayList<>(); 5 6 for(int num: A) { 7 val = ((val << 1) | num)%5; 8 divisible.add(val == 0); 9 } 10 11 return divisible; 12 } 13 }
原文地址:https://www.cnblogs.com/taste-it-own-it-love-it/p/10652723.html
时间: 2024-10-16 13:00:30