http://arc073.contest.atcoder.jp/tasks/arc073_d
dp高速化。
dp[i][j]:=(i番目のqueryまで処理して、2点の座標がx[i]とjの時の最小値)とする。
するとdp[i][j]+jとdp[i][j]-jの値についてrange_sumとrange_minが出来れば高速に更新できる。
2つsegtreeを持たないといけなくて、いろいろバグらせた。
ちゃんと式書いて区間をしっかり考えよう。
struct node { ll sum, lazy; node() : sum(0), lazy(0){} node(ll sum, ll lazy) : sum(sum), lazy(lazy){} }; int N; node seg1[200000 * 4], seg2[200000 * 4]; inline void lazy_eval(node* seg, int k, int l, int r) { if(seg[k].lazy) { seg[k * 2 + 1].sum += seg[k].lazy;//if range_min, erase (m - l) seg[k * 2 + 1].lazy += seg[k].lazy; seg[k * 2 + 2].sum += seg[k].lazy;//here too seg[k * 2 + 2].lazy += seg[k].lazy; seg[k].lazy = 0; } } void update(node* seg, int a, int b, ll s, int k = 0, int l = 0, int r = N) { if(b <= l || r <= a) return; if(a <= l && r <= b) { seg[k].sum += s;//here too seg[k].lazy += s; } else { int m = (l + r) / 2; lazy_eval(seg, k, l, r); update(seg, a, b, s, k * 2 + 1, l, m); update(seg, a, b, s, k * 2 + 2, m, r); seg[k].sum = min(seg[k * 2 + 1].sum, seg[k * 2 + 2].sum); } } ll get_sum(node* seg, int a, int b, int k = 0, int l = 0, int r = N) { if(b <= l || r <= a) return inf * 2; if(a <= l && r <= b) return seg[k].sum; else { int m = (l + r) / 2; lazy_eval(seg, k, l, r); ll lsum = get_sum(seg, a, b, k * 2 + 1, l, m); ll rsum = get_sum(seg, a, b, k * 2 + 2, m, r); return min(lsum, rsum); } } int Q, A, B; int C[MAX_N]; void solve() { cin >> N >> Q >> A >> B; A--; B--; C[0] = B; rep(i, 1, Q + 1) { cin >> C[i]; C[i]--; } rep(i, 0, N) { if(i != A) { update(seg1, i, i + 1, inf - i); update(seg2, i, i + 1, inf + i); } else { update(seg1, i, i + 1, -i); update(seg2, i, i + 1, i); } } rep(i, 1, Q + 1) { ll m = get_sum(seg1, 0, C[i] + 1) + C[i]; MIN(m, get_sum(seg2, C[i], N) - C[i]); update(seg1, 0, N, abs(C[i] - C[i - 1])); update(seg2, 0, N, abs(C[i] - C[i - 1])); ll r = get_sum(seg1, C[i - 1], C[i - 1] + 1) + C[i - 1]; if(r > m) { update(seg1, C[i - 1], C[i - 1] + 1, m - r); update(seg2, C[i - 1], C[i - 1] + 1, m - r); } } ll res = inf; rep(i, 0, N) { MIN(res, get_sum(seg1, i, i + 1) + i); } cout << res << "\n"; }