n개의 요소가 있는 배열 A와 q개의 쿼리가 있는 쿼리 Q의 다른 목록이 있다고 가정합니다. eachQuery[i]는 (x, k) 쌍을 포함합니다. 쿼리를 처리할 때 x의 경우 A[x] 값을 1만큼 줄입니다. k의 경우 k번째로 큰 요소를 인쇄합니다. 처음에 A의 모든 요소는 0 또는 1입니다.
따라서 입력이 A =[1, 1, 0, 1, 0]과 같으면; Q =[[2, 3], [1, 2], [2, 3], [2, 1], [2, 5]], 출력은 [1, 1, 1, 0]
단계
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
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
예시
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
#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); }
입력
{ 1, 1, 0, 1, 0 }, { { 1, 2 }, { 0, 1 }, { 1, 2 }, { 1, 0 }, { 1, 4 }}
출력
1, 1, 1, 0,