leetcode 898.子数组按位或操作 Java

题目链接

https://leetcode-cn.com/problems/bitwise-ors-of-subarrays/

描述

我们有一个非负整数数组 A。

对于每个（连续的）子数组 B = [A[i], A[i+1], ..., A[j]] （ i <= j），我们对 B 中的每个元素进行按位或操作，获得结果 A[i] | A[i+1] | ... | A[j]。

返回可能结果的数量。 （多次出现的结果在最终答案中仅计算一次。）

提示：

1 <= A.length <= 50000
0 <= A[i] <= 10^9

示例

示例 1：

输入：[0]
输出：1
解释：
只有一个可能的结果 0 。

示例 2：

输入：[1,1,2]
输出：3
解释：
可能的子数组为 [1]，[1]，[2]，[1, 1]，[1, 2]，[1, 1, 2]。
产生的结果为 1，1，2，1，3，3 。
有三个唯一值，所以答案是 3 。

示例 3：

输入：[1,2,4]
输出：6
解释：
可能的结果是 1，2，3，4，6，以及 7 。

初始代码模板

class Solution {
    public int subarrayBitwiseORs(int[] A) {

    }
}

代码

class Solution {
    public int subarrayBitwiseORs(int[] A) {
        Set<Integer> ans = new HashSet<>();
        Set<Integer> cur = new HashSet<>();
        cur.add(0);

        for (int a : A) {
            Set<Integer> set = new HashSet<>();

            for (int num : cur) {
                set.add(num | a);
                ans.add(num | a);
            }
            set.add(a);
            ans.add(a);

            cur = set;
        }

        return ans.size();
    }
}