Advertisement
Josif_tepe

Untitled

Dec 22nd, 2024
101
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.15 KB | None | 0 0
  1. #include <iostream>
  2. #include <queue>
  3. using namespace std;
  4.  
  5.  
  6. int main()
  7. {
  8.     int T;
  9.     cin >> T;
  10.    
  11.     int di[] = {-1, 1, 0, 0};
  12.     int dj[] = {0, 0, -1, 1};
  13.    
  14.     for(int t = 0; t < T; t++) {
  15.         int n, m;
  16.         cin >> n >> m;
  17.        
  18.         char mat[n][m];
  19.         int si, sj;
  20.         int ei, ej;
  21.         for(int i = 0; i < n; i++) {
  22.             for(int j = 0; j < m; j++) {
  23.                 cin >> mat[i][j];
  24.                
  25.                 if(mat[i][j] == 'S') {
  26.                     si = i;
  27.                     sj = j;
  28.                 }
  29.                 if(mat[i][j] == 'E') {
  30.                     ei = i;
  31.                     ej = j;
  32.                 }
  33.             }
  34.         }
  35.         queue<int> q;
  36.         q.push(ei);
  37.         q.push(ej);
  38.         q.push(0);
  39.        
  40.         vector<vector<int>> dist(n, vector<int>(m, -1));
  41.         vector<vector<bool>> visited(n, vector<bool>(m, false));
  42.        
  43.         while(!q.empty()) {
  44.             int ci = q.front();
  45.             q.pop();
  46.             int cj = q.front();
  47.             q.pop();
  48.             int d = q.front();
  49.             q.pop();
  50.            
  51.             dist[ci][cj] = d;
  52.            
  53.             for(int k = 0; k < 4; k++) {
  54.                 int ti = ci + di[k];
  55.                 int tj = cj + dj[k];
  56.                
  57.                 if(ti >= 0 and ti < n and tj >= 0 and tj < m and mat[ti][tj] != 'T' and !visited[ti][tj]) {
  58.                     q.push(ti);
  59.                     q.push(tj);
  60.                     q.push(d + 1);
  61.                     visited[ti][tj] = true;
  62.                 }
  63.             }
  64.         }
  65.        
  66.         int najkratok_pat_od_S_do_E = dist[si][sj];
  67.         int health = 100;
  68.         for(int i = 0; i < n; i++) {
  69.             for(int j = 0; j < m; j++) {
  70.                 if(isdigit(mat[i][j]) and dist[i][j] != -1) {
  71.                     if(dist[i][j] < najkratok_pat_od_S_do_E) {
  72.                         health -= (mat[i][j] - '0');
  73.                     }
  74.                 }
  75.             }
  76.         }
  77.         if(health < 0) {
  78.             health = 0;
  79.         }
  80.         cout << health << endl;
  81.        
  82.     }
  83.     return 0;
  84.  
  85. }
  86.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement