檢查查詢 0 和的 C++ 程式碼


假設我們有一個 n 元素陣列 A,陣列中的元素在 -1 到 1 的範圍內。另一個包含 m 個查詢的陣列 Q,其中 Q[i] = (li, ri)。如果陣列 a 的元素可以重新排列,使得 Q[li] + ... + Q[ri] = 0,則查詢的響應為 1,否則為 0。我們必須找到所有查詢的答案。

因此,如果輸入為 A = [-1, 1, 1, 1, -1];Q = [[1, 1], [2, 3], [3, 5], [2, 5], [1, 5]],則輸出為 [0, 1, 0, 1, 0]

步驟

為了解決此問題,我們將按照以下步驟進行 -

n := size of A
m := size of Q
z := 0
for initialize , i := 0, when i < n, update (increase i by 1), do:
   z := z + (1 if a < 0, otherwise 0)
if z > n - z, then:
   z := n - z
for initialize i := 0, when i < m, update (increase i by 1), do:
   l := Q[i, 0]
   r := Q[i, 1]
   print 1 if (((r - l) mod 2 is 1 and (r - l + 1) / 2) <= z),
otherwise 0

示例

讓我們看看下面的實現,以便更好地理解 -

#include <bits/stdc++.h>
using namespace std;
void solve(vector<int> A, vector<vector<int>> Q){
   int n = A.size();
   int m = Q.size();
   int z = 0;
   for (int a, i = 0; i < n; ++i)
      z += a < 0;
   if (z > n - z)
      z = n - z;
   for (int i = 0; i < m; i++){
      int l = Q[i][0];
      int r = Q[i][1];
      cout << (((r - l) % 2 && (r - l + 1) / 2) <= z) << ", ";
   }
}
int main(){
   vector<int> A = { -1, 1, 1, 1, -1 };
   vector<vector<int>> Q = { { 1, 1 }, { 2, 3 }, { 3, 5 }, { 2, 5 }, { 1, 5 } };
   solve(A, Q);
}

輸入

{ -1, 1, 1, 1, -1 }, { { 1, 1 }, { 2, 3 }, { 3, 5 }, { 2, 5 }, { 1, 5
} }

輸出

1, 0, 1, 0, 1,

更新於:2022 年 3 月 30 日

124 次瀏覽

開啟您的 職業 生涯

完成課程,獲得認證

開始
廣告