leetcode898.子数组按位或操作

题目大意

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

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

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

示例 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 。

解题思路

或操作是单调不减的，因此首先对整个数组进行或操作，找到最大值mask。
然后两层遍历，计算每个区间的或操作结果，若当前结果已经等于mask了，则后续元素或操作仍为mask，因此可以终止内层遍历。

class Solution {
public:
    int subarrayBitwiseORs(vector<int>& A) {
    	unordered_set<int> myset;
    	int mask = 0, length = A.size();

    	// 由于是或操作，每次的结果是单调不减的，首先找到最大结果
    	for (int i = 0; i < length; ++i)
    		mask |= A[i];

    	for (int i = length - 1; i >= 0; --i){
    		int tmp = 0;
    		for (int j = i; j < length; ++j){
    			tmp |= A[j];
    			if (tmp == mask)
    				break;
    			myset.insert(tmp);
    		}
    	}
    	return myset.size();
    }
};