Advertisement
Josif_tepe

Untitled

Mar 13th, 2023
876
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.26 KB | None | 0 0
  1. #include <iostream>
  2. #include <map>
  3. #include <cmath>
  4. #include <set>
  5. #include <vector>
  6. #include <queue>
  7. #include <algorithm>
  8. #include <cstring>
  9. #include <fstream>
  10. using namespace std;
  11. typedef long long ll;
  12. char mat[1005][1005];
  13. bool visited[1005][1005];
  14. int dist[1005][1005];
  15. int main() {
  16.     ios_base::sync_with_stdio(false);
  17.     int n, m, k;
  18.     cin >> n >> m >> k;
  19.     int si, sj, ei, ej;
  20.     for(int i = 0; i < n; i++) {
  21.         for(int j = 0; j < m; j++) {
  22.             cin >> mat[i][j];
  23.             if(mat[i][j] == 'M') {
  24.                 si = i;
  25.                 sj = j;
  26.             }
  27.             if(mat[i][j] == 'V') {
  28.                 ei = i;
  29.                 ej = j;
  30.             }
  31.         }
  32.     }
  33.     queue<int> q;
  34.     q.push(si);
  35.     q.push(sj);
  36.     memset(visited, false, sizeof visited);
  37.     visited[si][sj] = true;
  38.     int di[] = {-1, 1, 0, 0};
  39.     int dj[] = {0, 0, -1, 1};
  40.     vector<pair<int, int> > cell_at_border;
  41.     while(!q.empty()) {
  42.         int ci = q.front(); q.pop();
  43.         int cj = q.front(); q.pop();
  44.         if(ci == ei and cj == ej) {
  45.             cout << "DA" << endl;
  46.             return 0;
  47.         }
  48.         for(int i = 0; i < 4; i++) {
  49.             int ti = ci + di[i];
  50.             int tj = cj + dj[i];
  51.            
  52.             if(ti >= 0 and ti < n and tj >= 0 and tj < m and !visited[ti][tj]) {
  53.                 if(mat[ti][tj] == '#') {
  54.                     cell_at_border.push_back(make_pair(ci, cj));
  55.                 }
  56.                 else {
  57.                     q.push(ti);
  58.                     q.push(tj);
  59.                     visited[ti][tj] = true;
  60.                 }
  61.             }
  62.         }
  63.     }
  64.     memset(visited, false, sizeof visited);
  65.     memset(dist, -1, sizeof dist);
  66.     for(int i = 0; i < cell_at_border.size(); i++) {
  67.         q.push(cell_at_border[i].first);
  68.         q.push(cell_at_border[i].second);
  69.         q.push(0);
  70.         visited[cell_at_border[i].first][cell_at_border[i].second] = true;
  71.         dist[cell_at_border[i].first][cell_at_border[i].second] = 0;
  72.     }
  73.    
  74.     while(!q.empty()) {
  75.         int ci = q.front(); q.pop();
  76.         int cj = q.front(); q.pop();
  77.         int d = q.front(); q.pop();
  78.         dist[ci][cj] = d;
  79.         for(int i = 0; i < 4; i++) {
  80.             int ti = ci + di[i];
  81.             int tj = cj + dj[i];
  82.             if(ti >= 0 and ti < n and tj >= 0 and tj < m and !visited[ti][tj]) {
  83.                 visited[ti][tj] = true;
  84.                 q.push(ti);
  85.                 q.push(tj);
  86.                 q.push(d + 1);
  87.             }
  88.         }
  89.        
  90.     }
  91.     memset(visited, false, sizeof visited);
  92.     q.push(ei);
  93.     q.push(ej);
  94.     while(!q.empty()) {
  95.         int ci = q.front(); q.pop();
  96.         int cj = q.front(); q.pop();
  97.        
  98.             if(dist[ci][cj] <= k) {
  99.                 cout << "DA" << endl;
  100.                 return 0;
  101.             }
  102.         for(int i = 0; i < 4; i++) {
  103.             int ti = ci + di[i];
  104.             int tj = cj + dj[i];
  105.             if(ti >= 0 and ti < n and tj >= 0 and tj < m and !visited[ti][tj] and mat[ti][tj] != '#') {
  106.                 visited[ti][tj] = true;
  107.                 q.push(ti);
  108.                 q.push(tj);
  109.             }
  110.         }
  111.     }
  112.     cout << "NE" << endl;
  113.     return 0;
  114. }
  115. /*
  116.  M..
  117.  ###
  118.  ..V
  119.  **/
  120.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement