Mã C ++ để xử lý hoạt động truy vấn trên mảng nhị phân

Giả sử chúng ta có một mảng A với n phần tử và một danh sách các truy vấn Q với q truy vấn. eachQuery [i] chứa một cặp (x, k). Khi chúng ta xử lý một truy vấn, đối với x:giảm giá trị của A [x] đi 1. Đối với k, in ra phần tử lớn thứ k. Ban đầu tất cả các phần tử trong A là 0 hoặc 1.

Vì vậy, nếu đầu vào là A =[1, 1, 0, 1, 0]; Q =[[2, 3], [1, 2], [2, 3], [2, 1], [2, 5]], thì đầu ra sẽ là [1, 1, 1, 0]

Các bước

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

n := size of A
m := 0
for initialize i := 0, when i < n, update (increase i by 1), do:
   if A[i] is non-zero, then:
      (increase m by 1)
for initialize j := 0, when j < size of Q, update (increase j by 1),
do:
   x := Q[j, 0]
   k := Q[j, 1]
   if x is same as 0, then:
      if A[k] is non-zero, then:
         (decrease m by 1)
      Otherwise
         (increase m by 1)
      A[k] := A[k] XOR 1
   Otherwise
      if m >= k, then:
         print 1
      Otherwise
         print 0

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

#include bits/stdc++.h>
using namespace std;
void solve(vector<int> A, vector<vector<int>> Q){
   int n = A.size();
   int m = 0;
   for (int i = 0; i < n; i++){
      if (A[i])
      m++;
   }
   for (int j = 0; j < Q.size(); j++){
      int x = Q[j][0];
      int k = Q[j][1];
      if (x == 0){
         if (A[k])
            m--;
         else
            m++;
         A[k] ^= 1;
      }
      else{
         if (m >= k)
            cout << 1 << ", ";
         else
            cout << 0 << ", ";
      }
   }
}
int main(){
   vector<int> A = { 1, 1, 0, 1, 0 };
   vector<vector<int>> Q = { { 1, 2 }, { 0, 1 }, { 1, 2 }, { 1, 0 },{ 1, 4 } };
   solve(A, Q);
}

Đầu vào

{ 1, 1, 0, 1, 0 }, { { 1, 2 }, { 0, 1 }, { 1, 2 }, { 1, 0 }, { 1, 4 }}

Đầu ra

1, 1, 1, 0,