#include <bits/stdc++.h>
#include "walk.h"
#include "grader.cpp"
using namespace std;
//macros
typedef long long ll;
typedef pair<int, int> ii;
typedef pair<ll, ll> lll;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
typedef vector<lll> vlll;
#define REP(a,b,c) for(int a=int(b); a<int(c); a++)
#define RE(a,c) REP(a,0,c)
#define RE1(a,c) REP(a,1,c+1)
#define REI(a,b,c) REP(a,b,c+1)
#define REV(a,b,c) for(int a=int(c-1); a>=int(b); a--)
#define FOR(a,b) for(auto& a:b)
#define INF 1e18
#define pb push_back
#define fi first
#define se second
#define sz size()
#define all(a) a.begin(), a.end()
ll min_distance(vi x, vi h, vi l, vi r, vi y, int s, int g) {
l.pb(s); r.pb(s); y.pb(0);
l.pb(g); r.pb(g); y.pb(0);
int n=x.sz, m=l.sz, N=0;
ll ans = INF;
map<ii, int> vertexes;
map<int, vi> horVertex;
vector<vii> adj;
auto getVert = [&](int x, int y) {
if(vertexes.count({x,y})) return vertexes[{x,y}];
horVertex[y].pb(x);
return vertexes[{x,y}] = N++;
};
auto connect = [&](ii a, ii b) {
int u = getVert(a.fi, a.se);
int v = getVert(b.fi, b.se);
int d = abs(a.fi-b.fi)+abs(a.se-b.se);
adj[u].pb({v,d});
adj[v].pb({u,d});
};
adj.resize(m*5);
// create vertexes, and vertical edges
{
priority_queue<iii> pq;
set<int> st;
RE(i,m) pq.push({r[i],2,i});
RE(i,m) pq.push({r[i],1,i});
RE(i,m) pq.push({l[i],1,i});
RE(i,m) pq.push({l[i],0,i});
while(!pq.empty()) {
iii p = pq.top(); pq.pop();
int cx, t, i;
tie(cx,t,i) = p;
if(t == 2) {
// right
st.insert(y[i]);
} else if(t == 1) {
// connect
getVert(x[cx], y[i]);
auto it = st.lower_bound(y[i]);
if(it != st.begin())
--it, connect({x[cx],y[i]}, {x[cx],*it});
} else {
// left
st.erase(y[i]);
}
}
}
// create horizontal edges
{
FOR(p,horVertex) sort(all(p.se));
RE(i,m) {
if(l[i] == r[i]) continue;
vi& f = horVertex[y[i]];
auto it=lower_bound(all(f), x[l[i]]);
auto ed=upper_bound(all(f), x[r[i]]);
while(it != ed) {
auto prev = it++;
if(it == ed) break;
connect({*prev, y[i]}, {*it, y[i]});
}
}
}
// dijkstra
{
vll dist; dist.assign(N, INF);
priority_queue<lll> pq;
dist[getVert(x[s],0)] = 0; pq.push({getVert(x[s],0), 0});
while(!pq.empty()) {
lll p = pq.top(); pq.pop();
int u=p.fi; ll d=p.se;
if(dist[u] != d) continue;
FOR(p,adj[u]) {
int v = p.fi;
ll add = p.se;
if(d+add < dist[v]) {
dist[v] = d+add;
pq.push({v,dist[v]});
}
}
}
ans = dist[getVert(x[g],0)];
}
if(ans == INF) ans = -1;
return ans;
}
Compilation message
walk.cpp: In function 'll min_distance(vi, vi, vi, vi, vi, int, int)':
walk.cpp:32:9: warning: unused variable 'n' [-Wunused-variable]
32 | int n=x.sz, m=l.sz, N=0;
| ^
/tmp/ccyMoVJ8.o: In function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'
/tmp/ccJ9ccbx.o:walk.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
