Advertisement
Josif_tepe

Untitled

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