C++ 中的子数组按位或运算
假设我们有一个非负整数数组 A。对于每个(连续)子数组,比如 B = [A[i], A[i+1], ..., A[j]](i <= j),我们将执行 B 中所有元素的按位 OR,得到一个结果 A[i] | A[i+1] | ... | A[j]。我们需要找出可能的不同的结果数量。(答案中的重复结果只计算一次。)
因此,如果输入类似于 [1,1,2],那么结果将是 3,因为子数组是 [1]、[1]、[2]、[1,1]、[1,2]、[1,1,2],那么结果将是 1、1、2、1、3、3,因此有三个不同的结果。
要解决这个问题,我们将遵循以下步骤 −
创建两个集合 ret 和 curr2
对于 0 到数组长度中的 i
创建一个集合 curr1,其中插入 A[i]
对于 curr2 中的每个元素 e
将 (e OR A[i]) 插入 curr1
对于 curr1 中的每个元素 e
将 e 插入 ret
curr2 := curr1
返回 ret 的长度
示例
让我们看以下实现以获得更好的理解 −
#include <bits/stdc++.h> using namespace std; class Solution { public: int subarrayBitwiseORs(vector<int>& A) { unordered_set <int> ret; unordered_set <int> curr2; for(int i = 0; i < A.size(); i++){ unordered_set <int> curr1; curr1.insert(A[i]); unordered_set<int>::iterator it = curr2.begin(); while(it != curr2.end()){ curr1.insert(*it | A[i]); it++; } it = curr1.begin(); while(it != curr1.end()){ ret.insert(*it); it++; } curr2 = curr1; } return ret.size(); } }; main(){ vector<int> v = {1,1,2}; Solution ob; cout << (ob.subarrayBitwiseORs(v)); }
输入
[1,1,2]
输出
3
广告