Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

Bitwise OR ของ Subarrays ใน C++


สมมติว่าเรามีอาร์เรย์ A ของจำนวนเต็มไม่ติดลบ สำหรับอาร์เรย์ย่อย (ต่อเนื่องกัน) ทุกรายการ B =[A[i], A[i+1], ..., A[j]] (ด้วย i <=j) เราจะทำค่าระดับบิต OR ขององค์ประกอบทั้งหมดใน B ได้ผลลัพธ์ A[i] | A[i+1] | ... | อา[j]. เราต้องหาจำนวนผลลัพธ์ที่เป็นไปได้ (ผลลัพธ์ที่เกิดขึ้นมากกว่าหนึ่งครั้งจะถูกนับเพียงครั้งเดียวในคำตอบสุดท้าย)

ดังนั้นหากอินพุตเป็น [1,1,2] ผลลัพธ์จะเป็น 3 เนื่องจากอาร์เรย์ย่อยคือ [1], [1], [2], [1,1], [1,2], [1, 1,2] จากนั้นผลลัพธ์จะเป็น 1,1,2,1,3,3 จากนั้นจะมีผลลัพธ์ที่แตกต่างกันสามแบบ

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

  • สร้างสองชุด ret และ curr2

  • สำหรับฉันอยู่ในช่วง 0 ถึงขนาดของอาร์เรย์

    • ทำ setcurr1 ใส่ A[i] เข้าไป

    • สำหรับแต่ละองค์ประกอบ e ใน curr2 -

      • แทรก (e OR A[i]) ลงในสกุลเงิน1

    • สำหรับแต่ละองค์ประกอบ e curr1

      • ใส่ 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