898.我们有一个非负整数数组 A。
对于每个(连续的)子数组 B = [A[i], A[i+1], ..., A[j]] ( i <= j),我们对 B 中的每个元素进行按位或操作,获得结果 A[i] | A[i+1] | ... | A[j]。
返回可能结果的数量。 (多次出现的结果在最终答案中仅计算一次。)
可以发现,对于固定的 j,result(j, j), result(j - 1, j), result(j - 2), j, ..., result(1, j) 的值是单调不降的,因为将 result(k, j) 对 A[k - 1] 做按位或操作,得到的结果 result(k - 1, j) 一定不会变小。并且,根据按位或操作的性质,如果把 result(k, j) 和 result(k - 1, j) 都用二进制表示,那么后者将前者二进制表示中的若干个 0 变成了 1。
由于数组 A 中都是小于 10^9 的正整数,它们的二进制表示最多只有 32 位。因此从 result(j, j) 开始到 result(1, j) 结束,最多只会有 32 个 0 变成了 1,也就是说,result(j, j), result(j - 1, j), result(j - 2), j, ..., result(1, j) 中最多只有 32 个不同的数。这样我们就可以维护一个集合,存储所有以 j 为结尾的 result 值。当结尾从 j 枚举到 j + 1 时,我们将集合中的每个数对 A[j + 1] 做按位或操作,得到的新的 result 值也不会超过 32 个。
class Solution {
public int subarrayBitwiseORs(int[] A) {
Set<Integer> ans = new HashSet();
Set<Integer> cur = new HashSet();
cur.add(0);
for (int x: A) {
Set<Integer> cur2 = new HashSet();
for (int y: cur)
cur2.add(x | y);
cur2.add(x);
cur = cur2;
ans.addAll(cur);
}
return ans.size();
}
}