Advertisement
Josif_tepe

Untitled

Nov 25th, 2024
52
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.17 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <queue>
  4. #include <cmath>
  5. #include <map>
  6. using namespace std;
  7.  
  8. int main() {
  9.     int T;
  10.     cin >> T;
  11.    
  12.     int di[] = {-1, 1, 0, 0};
  13.     int dj[] = {0, 0, -1, 1};
  14.    
  15.     while(T--) {
  16.         int n, m;
  17.         cin >> n >> m;
  18.        
  19.         char mat[n][m];
  20.         int si, sj, 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.        
  36.         vector<vector<bool>> visited(n, vector<bool>(m, false));
  37.        
  38.         queue<int> q;
  39.         q.push(ei);
  40.         q.push(ej);
  41.         q.push(0);
  42.        
  43.         int distance_from_S_to_E = -1;
  44.         vector<pair<int, int>> wizards;
  45.         while(!q.empty()) {
  46.             int ci = q.front();
  47.             q.pop();
  48.             int cj = q.front();
  49.             q.pop();
  50.             int dist = q.front();
  51.             q.pop();
  52.            
  53.             if(ci == si and cj == sj) {
  54.                 distance_from_S_to_E = dist;
  55.             }
  56.             if(isdigit(mat[ci][cj])) {
  57.                 wizards.push_back(make_pair(dist, mat[ci][cj] - '0'));
  58.             }
  59.            
  60.             for(int i = 0; i < 4; i++) {
  61.                 int ti = ci + di[i];
  62.                 int tj = cj + dj[i];
  63.                
  64.                 if(ti >= 0 and ti < n and tj >= 0 and tj < m and !visited[ti][tj] and mat[ti][tj] != 'T') {
  65.                     q.push(ti);
  66.                     q.push(tj);
  67.                     q.push(dist + 1);
  68.                     visited[ti][tj] = true;
  69.                 }
  70.             }
  71.            
  72.         }
  73.        
  74.         int health = 100;
  75.         for(int i = 0; i < wizards.size(); i++) {
  76.             if(wizards[i].first <= distance_from_S_to_E) {
  77.                 health -= wizards[i].second;
  78.             }
  79.         }
  80.         cout << max(0, health) << endl;
  81.        
  82.        
  83.     }
  84.     return 0;
  85. }
  86.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement