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

更新日期:2020 年 4 月 30 日

741 次浏览

启动您的职业生涯

通过完成课程获得认证

开始
广告