Problem p03725 (C++)

Buggy submission (status: Time Limit Exceeded):
#include <algorithm>
#include <bitset>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <unordered_set>
#include <vector>

using namespace std;
using ll = long long;
using ull = unsigned long long;
// const ll mod = 998244353;
const ll mod = 1000000007;
const int inf = 1e9;
const ll linf = 1e18;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};

#define REP(i, n) for (ll i = 0; i < (n); ++i)
#define REP_FROM(i, j, n) for (ll i = (j); i < (n); ++i)
#define REP_REV(i, n) for (ll i = n - 1; i >= 0; --i)
#define REP_FROM_REV(i, j, n) for (ll i = n - 1; i >= j; --i)
#define all(x) (x).begin(), (x).end()
#define sz(x) ll(x.size())

template <typename T> inline T chmax(T &a, const T b) {
  return a = (a < b) ? b : a;
}

template <typename T> inline T chmin(T &a, const T b) {
  return a = (a > b) ? b : a;
}

ll power(ll base, ll exponent) {
  if (exponent % 2) {
    return power(base, exponent - 1) * base % mod;
  } else if (exponent) {
    ll root_ans = power(base, exponent / 2);
    return root_ans * root_ans % mod;
  } else {
    return 1;
  }
}

ll inverse(ll x) { return power(x, mod - 2); }

ll gcd(ll a, ll b) {
  if (a < b)
    gcd(b, a);
  ll r;
  while (r = a % b) {
    a = b;
    b = r;
  }
  return b;
}

template <typename T> ll sum(T begin, T end) {
  return accumulate(begin, end, 0ll);
}

struct combination {
  vector<ll> fact, inv;

  combination(int sz) : fact(sz + 1), inv(sz + 1) {
    fact[0] = 1;
    for (int i = 1; i <= sz; i++) {
      fact[i] = fact[i - 1] * i % mod;
    }
    inv[sz] = power(fact[sz], mod - 2);
    for (int i = sz - 1; i >= 0; i--) {
      inv[i] = inv[i + 1] * (i + 1) % mod;
    }
  }

  ll C(int p, int q) const {
    if (q < 0 || p < q)
      return 0;
    return (fact[p] * inv[q] % mod * inv[p - q] % mod);
  }
};

using Pair = pair<int, int>;
using LPair = pair<ll, ll>;

template <ll Modulus> struct ModInt {
  ll a;

  constexpr ModInt(const ll x = 0) noexcept : a((x % mod + mod) % mod) {}

  constexpr ll &value() noexcept { return a; }

  constexpr const ll &value() const noexcept { return a; }

  constexpr ModInt operator+(const ModInt x) const noexcept {
    return ModInt(*this) += x;
  }

  constexpr ModInt operator-(const ModInt x) const noexcept {
    return ModInt(*this) -= x;
  }

  constexpr ModInt operator*(const ModInt x) const noexcept {
    return ModInt(*this) *= x;
  }

  constexpr ModInt operator/(const ModInt x) const noexcept {
    return ModInt(*this) /= x;
  }

  constexpr ModInt &operator+=(const ModInt x) noexcept {
    a += x.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }

  constexpr ModInt &operator-=(const ModInt x) noexcept {
    if (a < x.a) {
      a += Modulus;
    }
    a -= x.a;
    return *this;
  }

  constexpr ModInt &operator*=(const ModInt x) noexcept {
    a = a * x.a % Modulus;
    return *this;
  }

  constexpr ModInt &operator/=(ModInt x) noexcept {
    ll exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= x;
      }
      x *= x;
      exp /= 2;
    }
    return *this;
  }

  constexpr ModInt operator-() noexcept { return ModInt(-a); }

  friend ostream &operator<<(ostream &os, const ModInt &m) {
    os << m.a;
    return os;
  }
};

using mint = ModInt<mod>;

int main() {
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(15);
  int h, w, k;
  cin >> h >> w >> k;
  vector<string> a(h);
  REP(i, h) cin >> a[i];
  int y, x;
  REP(i, h) {
    REP(j, w) {
      if (a[i][j] == 'S') {
        y = i;
        x = j;
      }
    }
  }
  vector<vector<int>> dp(h, vector<int>(w, inf));
  queue<Pair> q;
  q.emplace(y, x);
  dp[y][x] = 0;
  int maxY = y, minY = y, maxX = x, minX = x;
  while (q.size()) {
    auto cur = q.front();
    q.pop();
    REP(i, 4) {
      y = cur.first + dy[i];
      x = cur.second + dx[i];
      if (y < 0 || h <= y || x < 0 || w <= x)
        continue;
      if (dp[y][x] <= dp[cur.first][cur.second])
        continue;
      if (a[y][x] == '#')
        continue;
      dp[y][x] = dp[cur.first][cur.second] + 1;
      chmax(maxY, y);
      chmin(minY, y);
      chmax(maxX, x);
      chmin(minX, x);
      if (dp[cur.first][cur.second] != k - 1)
        q.emplace(y, x);
    }
  }
  int ans = inf;
  chmin(ans, (minY + k - 1) / k);
  chmin(ans, (minX + k - 1) / k);
  chmin(ans, (h - maxY - 1 + k - 1) / k);
  chmin(ans, (w - maxX - 1 + k - 1) / k);
  cout << ans + 1 << endl;
}


Fixed submission:
#include <algorithm>
#include <bitset>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <unordered_set>
#include <vector>

using namespace std;
using ll = long long;
using ull = unsigned long long;
// const ll mod = 998244353;
const ll mod = 1000000007;
const int inf = 1e9;
const ll linf = 1e18;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};

#define REP(i, n) for (ll i = 0; i < (n); ++i)
#define REP_FROM(i, j, n) for (ll i = (j); i < (n); ++i)
#define REP_REV(i, n) for (ll i = n - 1; i >= 0; --i)
#define REP_FROM_REV(i, j, n) for (ll i = n - 1; i >= j; --i)
#define all(x) (x).begin(), (x).end()
#define sz(x) ll(x.size())

template <typename T> inline T chmax(T &a, const T b) {
  return a = (a < b) ? b : a;
}

template <typename T> inline T chmin(T &a, const T b) {
  return a = (a > b) ? b : a;
}

ll power(ll base, ll exponent) {
  if (exponent % 2) {
    return power(base, exponent - 1) * base % mod;
  } else if (exponent) {
    ll root_ans = power(base, exponent / 2);
    return root_ans * root_ans % mod;
  } else {
    return 1;
  }
}

ll inverse(ll x) { return power(x, mod - 2); }

ll gcd(ll a, ll b) {
  if (a < b)
    gcd(b, a);
  ll r;
  while (r = a % b) {
    a = b;
    b = r;
  }
  return b;
}

template <typename T> ll sum(T begin, T end) {
  return accumulate(begin, end, 0ll);
}

struct combination {
  vector<ll> fact, inv;

  combination(int sz) : fact(sz + 1), inv(sz + 1) {
    fact[0] = 1;
    for (int i = 1; i <= sz; i++) {
      fact[i] = fact[i - 1] * i % mod;
    }
    inv[sz] = power(fact[sz], mod - 2);
    for (int i = sz - 1; i >= 0; i--) {
      inv[i] = inv[i + 1] * (i + 1) % mod;
    }
  }

  ll C(int p, int q) const {
    if (q < 0 || p < q)
      return 0;
    return (fact[p] * inv[q] % mod * inv[p - q] % mod);
  }
};

using Pair = pair<int, int>;
using LPair = pair<ll, ll>;

template <ll Modulus> struct ModInt {
  ll a;

  constexpr ModInt(const ll x = 0) noexcept : a((x % mod + mod) % mod) {}

  constexpr ll &value() noexcept { return a; }

  constexpr const ll &value() const noexcept { return a; }

  constexpr ModInt operator+(const ModInt x) const noexcept {
    return ModInt(*this) += x;
  }

  constexpr ModInt operator-(const ModInt x) const noexcept {
    return ModInt(*this) -= x;
  }

  constexpr ModInt operator*(const ModInt x) const noexcept {
    return ModInt(*this) *= x;
  }

  constexpr ModInt operator/(const ModInt x) const noexcept {
    return ModInt(*this) /= x;
  }

  constexpr ModInt &operator+=(const ModInt x) noexcept {
    a += x.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }

  constexpr ModInt &operator-=(const ModInt x) noexcept {
    if (a < x.a) {
      a += Modulus;
    }
    a -= x.a;
    return *this;
  }

  constexpr ModInt &operator*=(const ModInt x) noexcept {
    a = a * x.a % Modulus;
    return *this;
  }

  constexpr ModInt &operator/=(ModInt x) noexcept {
    ll exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= x;
      }
      x *= x;
      exp /= 2;
    }
    return *this;
  }

  constexpr ModInt operator-() noexcept { return ModInt(-a); }

  friend ostream &operator<<(ostream &os, const ModInt &m) {
    os << m.a;
    return os;
  }
};

using mint = ModInt<mod>;

int main() {
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(15);
  int h, w, k;
  cin >> h >> w >> k;
  vector<string> a(h);
  REP(i, h) cin >> a[i];
  int y, x;
  REP(i, h) {
    REP(j, w) {
      if (a[i][j] == 'S') {
        y = i;
        x = j;
      }
    }
  }
  vector<vector<int>> dp(h, vector<int>(w, inf));
  queue<Pair> q;
  q.emplace(y, x);
  dp[y][x] = 0;
  int maxY = y, minY = y, maxX = x, minX = x;
  while (q.size()) {
    auto cur = q.front();
    q.pop();
    REP(i, 4) {
      y = cur.first + dy[i];
      x = cur.second + dx[i];
      if (y < 0 || h <= y || x < 0 || w <= x)
        continue;
      if (dp[y][x] <= dp[cur.first][cur.second] + 1)
        continue;
      if (a[y][x] == '#')
        continue;
      dp[y][x] = dp[cur.first][cur.second] + 1;
      chmax(maxY, y);
      chmin(minY, y);
      chmax(maxX, x);
      chmin(minX, x);
      if (dp[cur.first][cur.second] != k - 1)
        q.emplace(y, x);
    }
  }
  int ans = inf;
  chmin(ans, (minY + k - 1) / k);
  chmin(ans, (minX + k - 1) / k);
  chmin(ans, (h - maxY - 1 + k - 1) / k);
  chmin(ans, (w - maxX - 1 + k - 1) / k);
  cout << ans + 1 << endl;
}