AtCoder Regular Contest 073F: Many Moves

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";
}