Advertisement
Josif_tepe

Untitled

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