检查查询 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,
广告