Lisäykset

input/code.cpp: In function 'int main()':
input/code.cpp:25:38: warning: suggest parentheses around comparison in operand of '^' [-Wparentheses]
   25 |     for (int j = 0; j < 4; j++) d(g[2<j^j] - (k - g[1] + 1) * (4%~j), (1^j)-j);
      |                                     ~^~

#include <bits/stdc++.h>
using namespace std;

#define N 101010
int n, m, fen[N];

int main() {
  ios_base::sync_with_stdio(0); cin.tie(0); cin>>n>>m;
  auto d = [&](int k, int x) { for (; k <= N; fen[k] += x, k += k & -k); };
  auto f = [&](int k, int s = 0) { for (; k >= 1; s += fen[k], k -= k & -k); return s; };
  auto b = [&](int l, int r, int t, int g) {
    while (l < r) {
      int mid = l + (r - l) / 2;
      if ((f(mid) == t) ^ g) r = mid;
      else l = mid + 1;
    }
    return l;
  };

  for (int i = 1, x = 0, p = 0; i <= n; i++, p = x) cin>>x, d(i, x-p), p = x;
  for (int q = 0, k = 0; q < m; q++) {
    cin>>k;
    int e = f(k);
    int g[3] = {1, b(1, k+1, e, 0), b(k, n+1, e, 1)};
    for (int j = 0; j < 4; j++) d(g[2<j^j] - (k - g[1] + 1) * (4%~j), (1^j)-j);
  }

  for (int i = 1; i <= n; i++) cout << f(i) << " ";
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 3

Verdict: ACCEPTED

Test 4

Group: 1, 3

Verdict: ACCEPTED

Test 5

Group: 1, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 3

Verdict: ACCEPTED

Test 8

Group: 3

Verdict: ACCEPTED

Test 9

Group: 3

Verdict: ACCEPTED

Test 10

Group: 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED