合計が0のクエリをチェックするC++コード

n個の要素を持つ配列Aがあり、要素が-1から1の範囲にあるとします。また、Q [i] =（li、ri）のようなm個のクエリQのペアの別の配列があります。配列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,