Advertisement
Josif_tepe

Untitled

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