This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
int main()
{
long long N;
cin >> N;
long long debx, deby, finx, finy;
cin >> debx >> deby >> finx >> finy;
debx--, deby--, finx--, finy--;
vector<long long> Tab(N);
map<long long, long long> M;
vector<long long> ys;
for (long long i = 0; i < N; i++)
{
cin >> Tab[i];
ys.push_back(Tab[i]);
}
if (N <= 1000)
{
ys.push_back(deby);
ys.push_back(finy);
long long buff = 0;
sort(ys.begin(), ys.end());
for (long long i = 0; i < ys.size(); i++)
M[ys[i]] = buff++;
deby = M[deby];
finy = M[finy];
for (long long i = 0; i < N; i++)
Tab[i] = M[Tab[i]];
vector<vector<long long>> occ(N, vector<long long>(N + 4));
priority_queue<vector<long long>> PQ;
PQ.push({0, debx, deby});
while (!PQ.empty())
{
long long x = PQ.top()[1], y = PQ.top()[2], t = PQ.top()[0];
PQ.pop();
if (occ[x][y])
continue;
occ[x][y] = 1;
if (x == finx && y == finy)
{
cout << -t;
return 0;
}
if (y > 0)
PQ.push({t - (ys[y] - ys[y - 1]), x, y - 1});
if (y < Tab[x])
PQ.push({t - (ys[y + 1] - ys[y]), x, y + 1});
if (x > 0 && y == 0)
PQ.push({t - 1, x - 1, Tab[x - 1]});
if (x < N - 1 && y == Tab[x])
PQ.push({t - 1, x + 1, 0});
if (x < N - 1)
PQ.push({t - 1, x + 1, min(y, Tab[x + 1])});
if (x > 0)
PQ.push({t - 1, x - 1, min(y, Tab[x - 1])});
}
}
else
{
long long minn = abs(deby - finy) + abs(debx - finx);
minn = min(minn, finy - 1 + N - finx + N - debx);
minn = min(minn, abs((debx - 1) - finx) + Tab[0] - finy + deby);
minn = min(minn, abs((debx + 1) - finx) + Tab[0] - deby + finy);
cout << minn;
}
}
Compilation message (stderr)
Main.cpp: In function 'int main()':
Main.cpp:24:33: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
24 | for (long long i = 0; i < ys.size(); i++)
| ~~^~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |