#include "rail.h"
#define N 5000
#define INF 0x3f3f3f3f
unsigned int X = 12345;
int rand_() {
return (X *= 3) >> 1;
}
int dd[N];
void sort(int *ii, int l, int r) {
while (l < r) {
int i = l, j = l, k = r, i_ = ii[l + rand_() % (r - l)], tmp;
while (j < k)
if (dd[ii[j]] == dd[i_])
j++;
else if (dd[ii[j]] < dd[i_]) {
tmp = ii[i], ii[i] = ii[j], ii[j] = tmp;
i++, j++;
} else {
k--;
tmp = ii[j], ii[j] = ii[k], ii[k] = tmp;
}
sort(ii, l, i);
l = k;
}
}
void findLocation(int n, int x0, int xx[], int tt[]) {
static int ii[N], jj[N];
int n1, n2, i, ir, d_, l, r;
d_ = INF, ir = -1;
for (i = 1; i < n; i++) {
dd[i] = getDistance(0, i);
if (d_ > dd[i])
d_ = dd[i], ir = i;
}
n1 = n2 = 0;
for (i = 0; i < n; i++) {
int d = getDistance(ir, i);
if (i == 0 || i != ir && d_ + d == dd[i])
dd[i] = d, ii[n1++] = i;
else
jj[n2++] = i;
}
tt[0] = 1, xx[0] = x0, tt[ir] = 2, xx[ir] = x0 + d_;
sort(ii, 0, n1);
l = ii[0], r = ir;
for (i = 1; i < n1; i++) {
int i_ = ii[i], d = getDistance(r, i_);
if (d == dd[i_] - (xx[ir] - xx[r]))
tt[i_] = 1, xx[i_] = xx[r] - d, l = i_;
else {
tt[i_] = 2, xx[i_] = xx[l] + (d - (xx[r] - xx[l]));
if (xx[r] > xx[i_])
r = i_;
}
}
sort(jj, 0, n2);
l = 0, r = jj[0];
for (i = 1; i < n2; i++) {
int j_ = jj[i], d = getDistance(l, j_);
if (d == dd[j_] - (xx[l] - xx[0]))
tt[j_] = 2, xx[j_] = xx[l] + d, r = j_;
else {
tt[j_] = 1, xx[j_] = xx[r] - (d - (xx[r] - xx[l]));
if (xx[l] < xx[j_])
l = j_;
}
}
}
