Advertisement
pb_jiang

abc241F WA

Oct 25th, 2022
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.16 KB | None | 0 0
  1. #include <assert.h>
  2. #include <bits/stdc++.h>
  3. using namespace std;
  4. #define dbg(...) logger(#__VA_ARGS__, __VA_ARGS__)
  5. template <typename... Args> void logger(string vars, Args &&... values)
  6. {
  7.     cerr << vars << " = ";
  8.     string delim = "";
  9.     (..., (cerr << delim << values, delim = ", "));
  10.     cerr << endl;
  11. }
  12.  
  13. template <class T> using mpq = priority_queue<T, vector<T>, greater<T>>;
  14. using ll = long long;
  15. using pii = pair<int, int>;
  16.  
  17. int h, w, n;
  18. int sx, sy, gx, gy;
  19. map<int, vector<int>> hm, vm;
  20.  
  21. int vcnt = 0;
  22. map<ll, int> v2i;
  23.  
  24. int get_id(int x, int y)
  25. {
  26.     ll id = ll(x) << 32 + y;
  27.     if (v2i.count(id))
  28.         return v2i[id];
  29.  
  30.     v2i[id] = vcnt++;
  31.     return v2i[id];
  32. }
  33.  
  34. struct Edge {
  35.     int v, next;
  36. } edges[3200003];
  37. int head[400003], ecnt = 0;
  38. void add_edge(int u, int v)
  39. {
  40.     if (u == v)
  41.         return;
  42.     edges[ecnt] = {v, head[u]};
  43.     head[u] = ecnt++;
  44. }
  45.  
  46. vector<pii> direct_add(int x, int y)
  47. {
  48.     vector<pii> ret;
  49.     int id = get_id(x, y);
  50.     if (hm.count(x)) {
  51.         const auto &ys = hm[x];
  52.         int ub = upper_bound(ys.begin(), ys.end(), y) - ys.begin();
  53.         if (ub < ys.size() && ys[ub] - y >= 1) {
  54.             int id1 = get_id(x, ys[ub] - 1), id2 = get_id(x, y);
  55.             add_edge(id2, id1);
  56.             ret.push_back({x, ys[ub] - 1});
  57.         }
  58.         if (ub > 0 && y - ys[ub - 1] >= 1) {
  59.             int id1 = get_id(x, ys[ub - 1] + 1), id2 = get_id(x, y);
  60.             add_edge(id2, id1);
  61.             ret.push_back({x, ys[ub - 1] + 1});
  62.         }
  63.     }
  64.     if (vm.count(y)) {
  65.         const auto &xs = vm[y];
  66.         int ub = upper_bound(xs.begin(), xs.end(), x) - xs.begin();
  67.         if (ub < xs.size() && xs[ub] - x >= 1) {
  68.             int id1 = get_id(xs[ub] - 1, y), id2 = get_id(x, y);
  69.             add_edge(id2, id1);
  70.             ret.push_back({xs[ub] - 1, y});
  71.         }
  72.         if (ub > 0 && x - xs[ub - 1] >= 1) {
  73.             int id1 = get_id(xs[ub - 1] + 1, y), id2 = get_id(x, y);
  74.             add_edge(id2, id1);
  75.             ret.push_back({xs[ub - 1] + 1, y});
  76.         }
  77.     }
  78.     return ret;
  79. }
  80.  
  81. int bfs()
  82. {
  83.     int s = get_id(sx, sy);
  84.     int t = get_id(gx, gy);
  85.  
  86.     vector<int> dist(400003, INT_MAX / 2);
  87.     dist[s] = 0;
  88.     queue<pii> q;
  89.     q.push({sx, sy});
  90.     while (q.empty() == false) {
  91.         auto h = q.front();
  92.         q.pop();
  93.         int x = h.first, y = h.second;
  94.         int u = get_id(x, y);
  95.         auto reach = direct_add(x, y);
  96.         for (const auto &p : reach) {
  97.             int v = get_id(p.first, p.second);
  98.             if (dist[v] > dist[u] + 1) {
  99.                 dist[v] = dist[u] + 1;
  100.                 q.push(p);
  101.             }
  102.         }
  103.     }
  104.     return dist[t] == INT_MAX / 2 ? -1 : dist[t];
  105. }
  106.  
  107. int main(int argc, char **argv)
  108. {
  109.     int x, y;
  110.     cin >> h >> w >> n;
  111.     cin >> sx >> sy;
  112.     cin >> gx >> gy;
  113.     memset(head, -1, sizeof(head));
  114.     for (int i = 0; i < n; ++i)
  115.         cin >> x >> y, hm[x].push_back(y), vm[y].push_back(x);
  116.     for (auto &p : hm)
  117.         sort(p.second.begin(), p.second.end());
  118.     for (auto &p : vm)
  119.         sort(p.second.begin(), p.second.end());
  120.  
  121.     cout << bfs() << endl;
  122.     return 0;
  123. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement