This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "seats.h"
#pragma GCC optimize ("Ofast")
#pragma GCC target ("avx2")
using namespace std;
using ll = long long;
using ld = long double;
#define ff first
#define ss second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define lb lower_bound
const int oo = 1e9 + 7;
int N, H, W, ST[4000005], cur[4000005], lazy[4000005];
vector<int> R, C, A;
void build(int v, int l, int r) {
if(l == r) {
ST[v] = 0; cur[v] = 1; return;
}
build(v * 2, l, (l + r) / 2);
build(v * 2 + 1, (l + r) / 2 + 1, r);
cur[v] = cur[v * 2] + cur[v * 2];
}
void prop(int v, int l, int r) {
if(l == r) return;
lazy[v * 2] += lazy[v];
lazy[v * 2 + 1] += lazy[v];
ST[v * 2] += lazy[v];
ST[v * 2 + 1] += lazy[v];
lazy[v] = 0;
}
void update(int v, int l, int r, int lo, int hi, int val) {
prop(v, l, r);
if(l > hi || r < lo) return;
if(l >= lo && r <= hi) {
ST[v] += val; lazy[v] += val;
prop(v, l, r); return;
}
update(v * 2, l, (l + r) / 2, lo, hi, val);
update(v * 2 + 1, (l + r) / 2 + 1, r, lo, hi, val);
ST[v] = min(ST[v * 2], ST[v * 2 + 1]); cur[v] = 0;
if(ST[v * 2] == ST[v]) cur[v] += cur[v * 2];
if(ST[v * 2 + 1] == ST[v]) cur[v] += cur[v * 2 + 1];
}
int calc(int i) {
int c = C[i], res = 0;
res += (c == 0 || A[c - 1] > i) ? 1 : -1;
res += (c == W - 1 || A[c + 1] > i) ? 1 : -1;
return res;
}
void give_initial_chart(int H, int W, vector<int> R, vector<int> C) {
::H = H, ::W = W, N = H * W;
::R = R, ::C = C; A.resize(N);
for(int l = 0; l < W; l++) A[C[l]] = l;
build(1, 0, N - 1);
for(int l = 0; l < W; l++) {
update(1, 0, N - 1, l, N - 1, calc(l));
}
}
void upd(int c, int val) {
if(c < 0 || c >= W) return;
update(1, 0, N - 1, A[c], N - 1, val * calc(c));
}
int swap_seats(int a, int b) {
for(int i = -1; i <= 1; i++) {
upd(C[a] + i, -1);
upd(C[b] + i, -1);
}
swap(C[a], C[b]);
swap(A[C[a]], A[C[b]]);
for(int i = -1; i <= 1; i++) {
upd(C[a] + i, 1);
upd(C[b] + i, 1);
}
return cur[1] - 1;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |