#include "rail.h"
#include<bits/stdc++.h>
using namespace std;
const int maxn = 5050;
const int inf = 1e9+10;
#define pb push_back
#define mp make_pair
#define fr first
#define sc second
#define all(x) x.begin(),x.end()
int d[maxn][maxn];
int p[maxn], tp[maxn];
int dist(int a, int b) {
if(d[a][b] != -1) return d[a][b];
return d[a][b] = getDistance(a,b);
}
void findLocation(int N, int first, int location[], int stype[]) {
int n = N;
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++) {
d[i][j] = -1;
}
}
p[0] = first;
tp[0] = 0;
// find the closest to 0
int b = -1;
for(int i = 1; i < n; i++) {
if(b == -1 || dist(0,i) < dist(0,b)) b = i;
}
p[b] = p[0]+dist(0,b);
tp[b] = 1;
// find the closest to b
// int a = -1;
// for(int i = 0; i < n; i++) {
// if(i == b) continue;
// if(a == -1 || dist(b,i) < dist(b,a)) a = i;
// }
// p[a] = p[b]-dist(b,a);
// tp[a] = 0;
int a = 0;
// location[a] = p[a]; stype[a] = tp[a]+1;
// location[b] = p[b]; stype[b] = tp[b]+1;
// for(int i = 0; i < n; i++) {
// if(i == a || i == b) continue;
// if(dist(a,i) <= dist(b,i)) {
// location[i] = p[a]+dist(a,i);
// stype[i] = 2;
// }
// else {
// location[i] = p[b]-dist(b,i);
// stype[i] = 1;
// }
// }
vector<pair<int,int>> lft,rgt;
for(int i = 0; i < n; i++) {
if(i == a || i == b) continue;
if(dist(a,i) < dist(b,i)) {
rgt.pb(mp(dist(a,i),i));
}
else {
lft.pb(mp(dist(b,i),i));
}
}
sort(all(rgt));
sort(all(lft));
for(int i = 0; i < rgt.size(); i++) {
int u = rgt[i].sc;
p[u] = -inf;
int v1 = -1;
for(int j = 0; j < i; j++) {
int v = rgt[j].sc;
// cout << u << " " << v << " " << 2*p[v]-p[a]-dist(a,u) << endl;
if(tp[v] == 1 && 2*p[v]-p[a]-dist(a,u) < p[v]) {
// p[u] = max(p[u],2*p[v]-p[a]-dist(a,u));
if(p[u] < 2*p[v]-p[a]-dist(a,u)) {
p[u] = 2*p[v]-p[a]-dist(a,u);
v1 = v;
}
}
}
if(v1 != -1 && dist(a,u) != dist(a,v1)+dist(v1,u)) {
p[u] = -inf;
}
if(p[u] == -inf) {
tp[u] = 1;
p[u] = p[a]+dist(a,u);
}
else {
tp[u] = 0;
}
}
for(int i = 0; i < lft.size(); i++) {
int u = lft[i].sc;
p[u] = inf;
int v1 = -1;
for(int j = 0; j < i; j++) {
int v = lft[j].sc;
if(tp[v] == 0 && 2*p[v]+dist(b,u)-p[b] > p[v]) {
if(p[u] > 2*p[v]+dist(b,u)-p[b]) {
p[u] = 2*p[v]+dist(b,u)-p[b];
v1 = v;
}
}
}
if(v1 != -1 && dist(b,u) != dist(b,v1)+dist(v1,u)) {
p[u] = inf;
}
if(p[u] == inf) {
tp[u] = 0;
p[u] = p[b]-dist(b,u);
}
else {
tp[u] = 1;
}
}
for(int i = 0; i < n; i++) {
location[i] = p[i];
stype[i] = tp[i]+1;
// cout << i << " - " << p[i] << " " << tp[i]+1 << endl;
}
}
