Advertisement
esraa_syam

graph

Sep 10th, 2023
867
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.84 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. #define nl "\n"
  3. #define fi first
  4. #define se second
  5. #define pi 3.14159
  6. #define ll long long
  7. #define odd(a)  (a&1)
  8. #define even(a)  !(a&1)
  9. #define Mod 1'000'000'007
  10. #define INF 2'000'000'000 // 2e9
  11. #define sz(x) int(x.size())
  12. #define charToInt(s) (s - '0')
  13. #define ull unsigned long long
  14. #define number_line iota(all(vec) , 1)
  15. #define all(s) s.begin(), s.end()
  16. #define rall(v) v.rbegin() , v.rend()
  17. #define fixed(n) fixed << setprecision(n)
  18. #define Num_of_Digits(n) ((int)log10(n) + 1)
  19. #define to_decimal(bin) stoll(bin, nullptr, 2)
  20. #define Ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  21. #define Floor(n, m) (((n) / (m)) - ((n) % (m) ? 0 : 1))
  22. #define Upper(s) transform(all(s), s.begin(), ::toupper);
  23. #define Lower(s) transform(all(s), s.begin(), ::tolower);
  24. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  25. //  ----- bits-------
  26. #define pcnt(n) __builtin_popcount(n)
  27. #define pcntll(n) __builtin_popcountll(n)
  28. #define clz(n) __builtin_clz(n)    // <---100
  29. #define clzll(n) __builtin_clzll(n)
  30. #define ctz(n) __builtin_ctz(n)  // 0001---->
  31. #define ctzll(n) __builtin_ctzll(n)
  32.  
  33. using namespace std;
  34.  
  35. template < typename T = int > istream& operator >> (istream &in, vector < T > & v){
  36.     for(auto & x : v) in >> x;
  37.     return in;
  38. }
  39.  
  40. template < typename T = int > ostream& operator << (ostream &out, const vector < T > & v){
  41.     for(const T & x : v) out << x << " ";
  42.     return out;
  43. }
  44.  
  45. void esraa()
  46. {
  47.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  48. #ifndef ONLINE_JUDGE
  49.     freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  50. #endif
  51. }
  52.  
  53. struct node{
  54.     int i , j , cnt;
  55.  
  56.     node(int i , int j , int cnt) : i(i) , j(j) , cnt(cnt) {}
  57. };
  58.  
  59. int dx[] = {0 , 0 , 1 , -1};
  60. int dy[] = {1 , -1 , 0 , 0};
  61. void solve(){
  62.     int n , m , d;
  63.     cin >> n >> m >> d;
  64.     vector < string > v(n);
  65.     for(auto & i : v) cin >> i;
  66.  
  67.     auto valid = [&](int x , int y){
  68.         return x >= 0 and x < n and y >= 0 and y < m and v[x][y] != 'M';
  69.     };
  70.  
  71.     int sx , sy , fx , fy;
  72.  
  73.     for(int i = 0; i < n; i++){
  74.         for(int j = 0; j < m; j++){
  75.             if(v[i][j] == 'S') sx = i , sy = j;
  76.             if(v[i][j] == 'F') fx = i , fy = j;
  77.         }
  78.     }
  79.  
  80.     auto bfs1 = [&](){
  81.         queue < node > q;
  82.  
  83.         for(int i = 0; i < n; i++){
  84.             for(int j = 0; j < m; j++){
  85.                 if(v[i][j] == 'M') q.push(node(i , j , d));
  86.             }
  87.         }
  88.  
  89.         while(!q.empty()){
  90.             auto [i , j , cnt] = q.front();
  91.             q.pop();
  92.  
  93.             if(cnt <= 0) continue;
  94.  
  95.             for(int k = 0; k < 4; k++){
  96.                 int x = i + dx[k];
  97.                 int y = j + dy[k];
  98.  
  99.                 if(valid(x , y)){
  100.                     v[x][y] = 'M';
  101.                     q.push(node(x , y , cnt - 1));
  102.                 }
  103.             }
  104.  
  105.  
  106.         }
  107.     };
  108.  
  109.     bfs1();
  110.  
  111.     auto bfs2 = [&](){
  112.         vector < vector < bool > > vis(n , vector < bool > (m));
  113.         queue < node > q;
  114.         for(int i = 0; i < n; i++){
  115.             for(int j = 0; j < m; j++){
  116.                 if(v[i][j] == 'S') q.push(node(i , j , 0));
  117.             }
  118.         }
  119.  
  120.         vis[sx][sy] = true;
  121.  
  122.         while(!q.empty()){
  123.             auto [i , j , cnt] = q.front();
  124.             q.pop();
  125.  
  126.             if(v[i][j] == 'F') return cnt;
  127.  
  128.             for(int k = 0; k < 4; k++){
  129.                 int x = i + dx[k];
  130.                 int y = j + dy[k];
  131.  
  132.                 if(valid(x , y) and !vis[x][y]){
  133.                     vis[x][y] = true;
  134.                     q.push(node(x , y , cnt + 1));
  135.                 }
  136.             }
  137.         }
  138.  
  139.         return -1;
  140.     };
  141.  
  142.     cout << bfs2() << nl;
  143.  
  144.  
  145. }
  146. int main()
  147. {
  148.     esraa();
  149.     int t = 1;
  150.     // cin >> t;
  151.     //cin.ignore();
  152.     while(t--)
  153.         solve();
  154.     return 0;
  155. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement