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 次瀏覽

啟動您的職業生涯

透過完成課程獲得認證

開始
廣告
© . All rights reserved.