# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
366839 | KoD | 철로 (IOI14_rail) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#ifndef __APPLE__
#include "rail.h"
#endif
template <class T>
using Vec = std::vector<T>;
void findLocation(int n, int first, int location[], int stype[]) {
assert(n <= 100);
Vec<Vec<int>> memo(n, Vec<int>(n, -1));
for (int i = 0; i < n; ++i) {
memo[i][i] = 0;
}
const auto ask = [&](const int i, const int j) {
if (memo[i][j] == -1) {
memo[i][j] = memo[j][i] = getDistance(i, j);
}
return memo[i][j];
};
int select = N;
for (int i = 1; i < n; ++i) {
if (select == N || ask(0, i) < ask(0, select)) {
select = i;
}
}
location[0] = first;
stype[0] = 1;
location[select] = location[0] + ask(0, select);
stype[select] = 2;
Vec<int> west, east;
for (int i = 1; i < n; ++i) {
if (i != select) {
if (ask(0, i) < ask(select, i)) {
east.push_back(i);
}
else {
west.push_back(i);
}
}
}
for (const auto i: east) {
location[i] = location[0] + ask(0, i);
stype[i] = 2;
}
for (const auto i: west) {
location[i] = location[select] - ask(select, i);
stype[i] = 1;
}
}