檢查查詢 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,
廣告