From ad80849894feb9e2530286262246d431d96303b2 Mon Sep 17 00:00:00 2001
From: Ren Baoshuo <i@baoshuo.ren>
Date: Fri, 19 Nov 2021 11:22:02 +0800
Subject: [PATCH] P4668 [BalticOI 2011 Day1]Treasures and Vikings

R62986744
---
 Luogu/P4668/P4668.cpp | 134 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100644 Luogu/P4668/P4668.cpp

diff --git a/Luogu/P4668/P4668.cpp b/Luogu/P4668/P4668.cpp
new file mode 100644
index 00000000..5a64ffb7
--- /dev/null
+++ b/Luogu/P4668/P4668.cpp
@@ -0,0 +1,134 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+const int to[4][2] = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
+
+int n, m, book[705][705];
+bool vis[705][705];
+char g[705][705];
+pair<int, int> y, v, t;
+
+struct node {
+    int x, y, step;
+
+    node()
+        : x(0), y(0), step(0) {}
+    node(int _x, int _y)
+        : x(_x), y(_y), step(0) {}
+    node(pair<int, int> _point)
+        : x(_point.first), y(_point.second), step(0) {}
+    node(pair<int, int> _point, int _step)
+        : x(_point.first), y(_point.second), step(_step) {}
+    node(int _x, int _y, int _step)
+        : x(_x), y(_y), step(_step) {}
+};
+
+/**
+ * @brief Preprocess the map
+ */
+void bfsV() {
+    memset(vis, 0x00, sizeof(vis));
+    memset(book, 0xff, sizeof(book));
+    queue<node> q;
+    q.push(v);
+    vis[v.first][v.second] = true;
+    book[v.first][v.second] = 0;
+    while (!q.empty()) {
+        int now;
+        auto u = q.front();
+        q.pop();
+
+        // Top
+        now = u.x - 1;
+        while (now >= 0 && g[now][u.y] != 'I') {
+            if (book[now][u.y] == -1) {
+                book[now][u.y] = u.step;
+            }
+            now--;
+        }
+
+        // Bottom
+        now = u.x + 1;
+        while (now < n && g[now][u.y] != 'I') {
+            if (book[now][u.y] == -1) {
+                book[now][u.y] = u.step;
+            }
+            now++;
+        }
+
+        // Left
+        now = u.y - 1;
+        while (now >= 0 && g[u.x][now] != 'I') {
+            if (book[u.x][now] == -1) {
+                book[u.x][now] = u.step;
+            }
+            --now;
+        }
+
+        // Right
+        now = u.y + 1;
+        while (now < m && g[u.x][now] != 'I') {
+            if (book[u.x][now] == -1) {
+                book[u.x][now] = u.step;
+            }
+            now++;
+        }
+
+        // Expand
+        for (int i = 0; i < 4; i++) {
+            node e = node(u.x + to[i][0], u.y + to[i][1], u.step + 1);
+            if (e.x >= 0 && e.x < n && e.y >= 0 && e.y < m && g[e.x][e.y] != 'I' && !vis[e.x][e.y]) {
+                vis[e.x][e.y] = true;
+                q.push(e);
+            }
+        }
+    }
+}
+
+/**
+ * @brief Search path to treasure
+ */
+void bfsY() {
+    memset(vis, 0x00, sizeof(vis));
+    queue<node> q;
+    q.push(node(y, 1));
+    while (!q.empty()) {
+        auto u = q.front();
+        q.pop();
+        for (int i = 0; i < 4; i++) {
+            node e = node(u.x + to[i][0], u.y + to[i][1], u.step + 1);
+            if (e.x >= 0 && e.x < n && e.y >= 0 && e.y < m && e.step <= book[e.x][e.y] && !vis[e.x][e.y]) {
+                vis[e.x][e.y] = true;
+                // Reached
+                if (e.x == t.first && e.y == t.second) {
+                    cout << "YES" << endl;
+                    return;
+                }
+                // Expand
+                q.push(e);
+            }
+        }
+    }
+    // Can't reach treasure
+    cout << "NO" << endl;
+}
+
+int main() {
+    // Read from stdin
+    cin >> n >> m;
+    for (int i = 0; i < n; i++) {
+        cin >> g[i];
+    }
+    // Mark you, viking and treasure
+    for (int i = 0; i < n; i++) {
+        for (int j = 0; j < m; j++) {
+            if (g[i][j] == 'Y') y = make_pair(i, j);
+            if (g[i][j] == 'V') v = make_pair(i, j);
+            if (g[i][j] == 'T') t = make_pair(i, j);
+        }
+    }
+    bfsV();
+    bfsY();
+    return 0;
+}