제출 #410971

#제출 시각아이디문제언어결과실행 시간메모리
410971Kevin_Zhang_TW고대 책들 (IOI17_books)C++17
50 / 100
147 ms8132 KiB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
void debug(auto l, auto r) { while (l != r) cerr << *l << " \n"[next(l)==r], ++l; }
#else
#define DE(...) 0
#define debug(...) 0
#endif
#include "books.h"

long long minimum_walk(std::vector<int> p, int s) {
	int n = p.size();
	ll res = 0;
	for (int i = 0;i < n;++i)
		res += abs(i - p[i]);

	if (res == 0) return 0;

	int L = 0, R = n-1;
	while (p[L] == L) ++L;
	while (p[R] == R) --R;

	chmax(R, s), chmin(L, s);

	auto expand = [&](int &l, int &r) {
		int nl = l, nr = r;
		auto upd = [&](int pos) {
			chmax(nr, p[pos]);
			chmin(nl, p[pos]);
		};
		upd(l), upd(r);

		while (nl != l || nr != r) {
			while (l > nl) {
				upd(--l);
			}
			while (r < nr) {
				upd(++r);
			}
		}
	};

	int l = s, r = s;
	expand(l, r);

	while (l != L || r != R) {
		int costl = 0, costr = 0;
		int nl = l, nr = r;

		bool nxt_l = false;
		{
			int sl = l, sr = r;
			while (sl > L && sr <= r) {
				expand(--sl, sr);
				costl += 2;
			}
			nxt_l = sr > r;
			chmin(nl, sl);
			chmax(nr, sr);
		}
		bool nxt_r = false;
		{
			int sl = l, sr = r;
			while (sl >= l && sr < R) {
				expand(sl, ++sr);
				costl += 2;
			}
			nxt_r = sl < l;
			chmin(nl, sl);
			chmax(nr, sr);
		}
		assert(nxt_r == nxt_l);
		if (nxt_r) res += min(costl, costr);
		else res += costl + costr;
		l = nl, r = nr;
	}

	return res;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...