oj.uz

답안 #491664

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
491664 2021-12-03T18:29:39 Z valerikk Climbers (RMI18_climbers) C++17
35 / 100
 800 ms 327820 KB 
#include <bits/stdc++.h>

using namespace std;

using ll = long long;

const int N = 5007;
const ll INF = 1e18;

int n;
int h[N];
ll d[N][N];

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);

	cin >> n;
	for (int i = 0; i < n; ++i) {
		cin >> h[i];
	}
	
	for (int i = 0; i < n; ++i) {
		for (int j = 0; j < n - 1; ++j) {
			d[i][j] = INF;
		}
	}

	priority_queue<pair<ll, pair<int, int>>, vector<pair<ll, pair<int, int>>>, greater<pair<ll, pair<int, int>>>> q;
	
	auto upd = [&](int i, int j, ll dd) {
		if (dd < d[i][j]) {
			d[i][j] = dd;
			q.push({dd, {i, j}});
		}
	};

	upd(0, n - 2, 0);

	while (!q.empty()) {
		auto z = q.top();
		q.pop();

		auto [i, j] = z.second;
	
		if (d[i][j] != z.first) {
			continue;
		}

		if (i == j || i - 1 == j) {
			cout << z.first << endl;
			return 0;
		}

		if (h[i] == h[j]) {
			if (j > 0) {
				upd(i, j - 1, d[i][j]);
			}
			upd(j, i, d[i][j]);
		}

		if (h[i] == h[j + 1]) {
			if (j < n - 1) {
				upd(i, j + 1, d[i][j]);
			}
			upd(j + 1, i, d[i][j]);
		}

		for (int ii : {i - 1, i + 1}) {
			if (ii < 0 || n <= ii) {
				continue;
			}

			if (h[i] < h[ii]) {
				if (h[i] < h[j]) {
					if (h[ii] - h[i] <= h[j] - h[i]) {
						upd(ii, j, d[i][j] + h[ii] - h[i]);
					} else {
						upd(j, min(ii, i), d[i][j] + h[j] - h[i]);
					}
				} else {
					if (h[ii] - h[i] <= h[j + 1] - h[i]) {
						upd(ii, j, d[i][j] + h[ii] - h[i]);
					} else {
						upd(j + 1, min(ii, i), d[i][j] + h[j + 1] - h[i]);
					}
				}
			}

			if (h[i] > h[ii]) {
				if (h[i] > h[j]) {
					if (-h[ii] + h[i] <= -h[j] + h[i]) {
						upd(ii, j, d[i][j] - h[ii] + h[i]);
					} else {
						upd(j, min(ii, i), d[i][j] - h[j] + h[i]);
					}
				} else {
					if (-h[ii] + h[i] <= -h[j + 1] + h[i]) {
						upd(ii, j, d[i][j] - h[ii] + h[i]);
					} else {
						upd(j + 1, min(ii, i), d[i][j] - h[j + 1] + h[i]);
					}
				}
			}	
		}
	}

	ll ans = INF;
	for (int i = 0; i < n - 1; ++i) {
		ans = min(ans, d[i][i]);
		ans = min(ans, d[i + 1][i]);
	}

	cout << ans << endl;
	return 0;
}

Subtask #1
25.0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 716 KB Output is correct
2 Correct 1 ms 2124 KB Output is correct
3 Correct 6 ms 12108 KB Output is correct
4 Correct 36 ms 82828 KB Output is correct
5 Correct 98 ms 196228 KB Output is correct

Subtask #2
10.0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 332 KB Output is correct
2 Incorrect 1 ms 588 KB Output isn't correct
3 Execution timed out 1098 ms 132532 KB Time limit exceeded
4 Correct 1 ms 716 KB Output is correct
5 Execution timed out 1094 ms 134692 KB Time limit exceeded

Subtask #3
0 / 50.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 1100 ms 132188 KB Time limit exceeded
2 Execution timed out 1093 ms 133700 KB Time limit exceeded
3 Execution timed out 1093 ms 143800 KB Time limit exceeded
4 Execution timed out 1096 ms 214456 KB Time limit exceeded
5 Execution timed out 1107 ms 214264 KB Time limit exceeded
6 Execution timed out 1099 ms 273044 KB Time limit exceeded
7 Execution timed out 1097 ms 273168 KB Time limit exceeded
8 Execution timed out 1098 ms 327756 KB Time limit exceeded
9 Execution timed out 1108 ms 327820 KB Time limit exceeded
10 Execution timed out 1097 ms 327812 KB Time limit exceeded