Problem p02644 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <algorithm>
#include <bits/stdc++.h>
#include <cctype>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <unordered_map>
#include <vector>

using namespace std;

typedef long double ld;
typedef long long ll;
typedef pair<ll, ll> pii;
typedef pair<ll, pii> piii;
typedef pair<pii, pii> piiii;
typedef pair<ll, bool> pib;

typedef vector<bool> vb;

typedef vector<ll> vi;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;

typedef vector<double> vf;

typedef queue<ll> qi;

typedef long double ld;

#define mp make_pair
#define mt make_tuple
#define pb push_back

#define rep(i, N) for (ll i = 0; i < N; i++)
#define repa(i, a, N) for (ll i = a; i < N; i++)
#define repr(i, N) for (ll i = N - 1; i >= 0; i--)

#define drep(i, N, j, M) rep(i, N) rep(j, M)
#define trep(i, N, j, M, k, L) rep(i, N) rep(j, M) rep(k, L)

#define all(A) A.begin(), A.end()
#define uni(A) A.erase(unique(all(A)), A.end())

template <class t, class u>
ostream &operator<<(ostream &os, const pair<t, u> &p) {
  return os << "{" << p.first << "," << p.second << "}";
}

template <class t> ostream &operator<<(ostream &os, const vector<t> &v) {
  os << "{";
  for (auto e : v)
    os << e << ",";
  return os << "}" << endl;
}

template <class t, class u>
ostream &operator<<(ostream &os, const unordered_map<t, u> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << "(" << itr->first << ", " << itr->second << "), ";
  }
  return os << "}";
}

template <class t> ostream &operator<<(ostream &os, const set<t> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << *itr << ", ";
  }
  return os << "}";
}

template <class t> ostream &operator<<(ostream &os, const multiset<t> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << *itr << ", ";
  }
  return os << "}";
}

const ll LLMAX = numeric_limits<long long>::max() / 3;
const double DMAX = numeric_limits<double>::max() / 3;

const double PI = 3.141592653589793;
// const ll MOD = 1e9 + 7;
const ll MOD = 998244353;

template <std::int_fast64_t Modulus> class modint {
  using i64 = std::int_fast64_t;

public:
  i64 a;

  constexpr modint(const i64 x = 0) noexcept : a(x % Modulus) {}
  constexpr i64 &value() noexcept { return a; }
  constexpr const i64 &value() const noexcept { return a; }
  constexpr modint operator+(const modint rhs) const noexcept {
    return modint(*this) += rhs;
  }
  constexpr modint operator-(const modint rhs) const noexcept {
    return modint(*this) -= rhs;
  }
  constexpr modint operator*(const modint rhs) const noexcept {
    return modint(*this) *= rhs;
  }
  constexpr modint operator/(const modint rhs) const noexcept {
    return modint(*this) /= rhs;
  }
  constexpr modint &operator+=(const modint rhs) noexcept {
    a += rhs.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }
  constexpr modint &operator-=(const modint rhs) noexcept {
    if (a < rhs.a) {
      a += Modulus;
    }
    a -= rhs.a;
    return *this;
  }
  constexpr modint &operator*=(const modint rhs) noexcept {
    a = a * rhs.a % Modulus;
    return *this;
  }
  constexpr modint &operator/=(modint rhs) noexcept {
    i64 exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= rhs;
      }
      rhs *= rhs;
      exp /= 2;
    }
    return *this;
  }
};

template <std::int_fast64_t Modulus>
ostream &operator<<(ostream &os, const modint<Modulus> &m) {
  return os << (m.a + Modulus) % Modulus;
}

typedef modint<MOD> mod;
typedef vector<mod> vm;
typedef vector<vm> vvm;
typedef vector<vvm> vvvm;

template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

void No() { cout << "No" << endl; }
void Yes() { cout << "Yes" << endl; }
void NO() { cout << "NO" << endl; }
void YES() { cout << "YES" << endl; }

vector<ll> fac(1e6 + 1);  // n!(mod M)
vector<ll> ifac(1e6 + 1); // k!^{M-2} (mod
ll mpow(ll x,
        ll n) { // x^n(mod M) ←普通にpow(x,n)では溢れてしまうため，随時mod計算
  ll ans = 1;
  while (n != 0) {
    if (n & 1)
      ans = ans * x % MOD;
    x = x * x % MOD;
    n = n >> 1;
  }
  return ans;
}

ll comb(ll a, ll b) { // aCbをmod計算
  if (a == 0 && b == 0)
    return 1;
  if (a < b || a < 0)
    return 0;
  ll tmp = ifac[a - b] * ifac[b] % MOD;
  return tmp * fac[a] % MOD;
}

void pre_calc_comb() {
  fac[0] = 1;
  ifac[0] = 1;
  for (ll i = 0; i < 1e6; i++) {
    fac[i + 1] = fac[i] * (i + 1) % MOD; // n!(mod M)
    ifac[i + 1] = ifac[i] * mpow(i + 1, MOD - 2) %
                  MOD; // k!^{M-2} (mod M) ←累乗にmpowを採用
  }
}

template <class T> class SegTree {
  int n;
  vector<T> data;
  T def;
  function<T(T, T)> operation;
  function<T(T, T)> update;
  T _query(int a, int b, int k, int l, int r) {
    if (r <= a || b <= l)
      return def;
    if (a <= l && r <= b)
      return data[k];
    else {
      T c1 = _query(a, b, 2 * k + 1, l, (l + r) / 2);
      T c2 = _query(a, b, 2 * k + 2, (l + r) / 2, r);
      return operation(c1, c2);
    }
  }

public:
  SegTree(size_t _n, T _def, function<T(T, T)> _operation,
          function<T(T, T)> _update)
      : def(_def), operation(_operation), update(_update) {
    n = 1;
    while (n < _n) {
      n *= 2;
    }
    data = vector<T>(2 * n - 1, def);
  }

  void change(int i, T x) {
    i += n - 1;
    data[i] = update(data[i], x);
    while (i > 0) {
      i = (i - 1) / 2;
      data[i] = operation(data[i * 2 + 1], data[i * 2 + 2]);
    }
  }

  T query(int a, int b) { return _query(a, b, 0, 0, n); }

  T operator[](int i) { return data[i + n - 1]; }
};

vector<ll> divisor(ll n) {
  vector<ll> ret;
  for (ll i = 1; i * i <= n; i++) {
    if (n % i == 0) {
      ret.push_back(i);
      if (i * i != n)
        ret.push_back(n / i);
    }
  }
  sort(begin(ret), end(ret));
  return (ret);
}

/*
SegTree<ll> st(N, LLMAX,
        // a: left child, b: right child
        [](ll a, ll b) {return min(a, b); },
        // d: old, x: query
        [](ll d, ll x) {return x; });
// */

ll gcd(ll a, ll b) {
  if (a == 0)
    return b;
  if (b == 0)
    return a;

  if (a % b == 0) {
    return (b);
  } else {
    return (gcd(b, a % b));
  }
}

ll N, M, K, Q;
ll ans = 0;
ll H, W;

void solve() {
  // N = 100; Q = 100;
  cin >> H >> W >> K;

  ll xs, ys, xe, ye;
  cin >> xs >> ys >> xe >> ye;
  xs--;
  ys--;
  xe--;
  ye--;

  vvi G(H, vi(W, 0));
  rep(h, H) {
    rep(w, W) {
      char c;
      cin >> c;
      G[h][w] = (c == '.');
    }
  }

  priority_queue<piiii> que;
  vvi D(H, vi(W, LLMAX));
  D[xs][ys] = 0;
  que.push(mp(mp(0, -1), mp(xs, ys)));

  while (que.empty() == false) {
    piiii p = que.top();
    que.pop();
    ll d = -p.first.first;
    ll n = p.first.second;
    ll x = p.second.first;
    ll y = p.second.second;

    auto fn = [&](ll n, ll k, ll x, ll y) {
      ll nx = x;
      ll ny = y;

      if (n) {
        nx += k;
      } else {
        ny += k;
      }

      if (nx < 0 || H <= nx || ny < 0 || W <= ny)
        return false;
      if (!G[nx][ny])
        return false;

      ll ad = 1 + (abs(k) - 1) / K;
      if (D[nx][ny] > d + ad) {
        D[nx][ny] = d + ad;
        que.push(mp(mp(-d - ad, !n), mp(nx, ny)));
        return true;
      } else if (D[nx][ny] == d + ad) {
        return true;
      }
      return false;
    };

    if (n == -1 || n) {
      repa(k, 1, H) {
        if (!fn(true, k, x, y))
          break;
      }
      repa(k, 1, H) {
        if (!fn(true, -k, x, y))
          break;
      }
    }
    if (n == -1 || !n) {
      repa(k, 1, W) {
        if (!fn(false, k, x, y))
          break;
      }

      repa(k, 1, W) {
        if (!fn(false, -k, x, y))
          break;
      }
    }
  }

  if (D[xe][ye] == LLMAX) {
    cout << -1 << endl;
    ;
  } else {
    cout << D[xe][ye] << endl;
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  solve();
  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <bits/stdc++.h>
#include <cctype>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <unordered_map>
#include <vector>

using namespace std;

typedef long double ld;
typedef long long ll;
typedef pair<ll, ll> pii;
typedef pair<ll, pii> piii;
typedef pair<pii, pii> piiii;
typedef pair<ll, bool> pib;

typedef vector<bool> vb;

typedef vector<ll> vi;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;

typedef vector<double> vf;

typedef queue<ll> qi;

typedef long double ld;

#define mp make_pair
#define mt make_tuple
#define pb push_back

#define rep(i, N) for (ll i = 0; i < N; i++)
#define repa(i, a, N) for (ll i = a; i < N; i++)
#define repr(i, N) for (ll i = N - 1; i >= 0; i--)

#define drep(i, N, j, M) rep(i, N) rep(j, M)
#define trep(i, N, j, M, k, L) rep(i, N) rep(j, M) rep(k, L)

#define all(A) A.begin(), A.end()
#define uni(A) A.erase(unique(all(A)), A.end())

template <class t, class u>
ostream &operator<<(ostream &os, const pair<t, u> &p) {
  return os << "{" << p.first << "," << p.second << "}";
}

template <class t> ostream &operator<<(ostream &os, const vector<t> &v) {
  os << "{";
  for (auto e : v)
    os << e << ",";
  return os << "}" << endl;
}

template <class t, class u>
ostream &operator<<(ostream &os, const unordered_map<t, u> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << "(" << itr->first << ", " << itr->second << "), ";
  }
  return os << "}";
}

template <class t> ostream &operator<<(ostream &os, const set<t> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << *itr << ", ";
  }
  return os << "}";
}

template <class t> ostream &operator<<(ostream &os, const multiset<t> &m) {
  os << "{";
  for (auto itr = m.begin(); itr != m.end(); ++itr) {
    os << *itr << ", ";
  }
  return os << "}";
}

const ll LLMAX = numeric_limits<long long>::max() / 3;
const double DMAX = numeric_limits<double>::max() / 3;

const double PI = 3.141592653589793;
// const ll MOD = 1e9 + 7;
const ll MOD = 998244353;

template <std::int_fast64_t Modulus> class modint {
  using i64 = std::int_fast64_t;

public:
  i64 a;

  constexpr modint(const i64 x = 0) noexcept : a(x % Modulus) {}
  constexpr i64 &value() noexcept { return a; }
  constexpr const i64 &value() const noexcept { return a; }
  constexpr modint operator+(const modint rhs) const noexcept {
    return modint(*this) += rhs;
  }
  constexpr modint operator-(const modint rhs) const noexcept {
    return modint(*this) -= rhs;
  }
  constexpr modint operator*(const modint rhs) const noexcept {
    return modint(*this) *= rhs;
  }
  constexpr modint operator/(const modint rhs) const noexcept {
    return modint(*this) /= rhs;
  }
  constexpr modint &operator+=(const modint rhs) noexcept {
    a += rhs.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }
  constexpr modint &operator-=(const modint rhs) noexcept {
    if (a < rhs.a) {
      a += Modulus;
    }
    a -= rhs.a;
    return *this;
  }
  constexpr modint &operator*=(const modint rhs) noexcept {
    a = a * rhs.a % Modulus;
    return *this;
  }
  constexpr modint &operator/=(modint rhs) noexcept {
    i64 exp = Modulus - 2;
    while (exp) {
      if (exp % 2) {
        *this *= rhs;
      }
      rhs *= rhs;
      exp /= 2;
    }
    return *this;
  }
};

template <std::int_fast64_t Modulus>
ostream &operator<<(ostream &os, const modint<Modulus> &m) {
  return os << (m.a + Modulus) % Modulus;
}

typedef modint<MOD> mod;
typedef vector<mod> vm;
typedef vector<vm> vvm;
typedef vector<vvm> vvvm;

template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

void No() { cout << "No" << endl; }
void Yes() { cout << "Yes" << endl; }
void NO() { cout << "NO" << endl; }
void YES() { cout << "YES" << endl; }

vector<ll> fac(1e6 + 1);  // n!(mod M)
vector<ll> ifac(1e6 + 1); // k!^{M-2} (mod
ll mpow(ll x,
        ll n) { // x^n(mod M) ←普通にpow(x,n)では溢れてしまうため，随時mod計算
  ll ans = 1;
  while (n != 0) {
    if (n & 1)
      ans = ans * x % MOD;
    x = x * x % MOD;
    n = n >> 1;
  }
  return ans;
}

ll comb(ll a, ll b) { // aCbをmod計算
  if (a == 0 && b == 0)
    return 1;
  if (a < b || a < 0)
    return 0;
  ll tmp = ifac[a - b] * ifac[b] % MOD;
  return tmp * fac[a] % MOD;
}

void pre_calc_comb() {
  fac[0] = 1;
  ifac[0] = 1;
  for (ll i = 0; i < 1e6; i++) {
    fac[i + 1] = fac[i] * (i + 1) % MOD; // n!(mod M)
    ifac[i + 1] = ifac[i] * mpow(i + 1, MOD - 2) %
                  MOD; // k!^{M-2} (mod M) ←累乗にmpowを採用
  }
}

template <class T> class SegTree {
  int n;
  vector<T> data;
  T def;
  function<T(T, T)> operation;
  function<T(T, T)> update;
  T _query(int a, int b, int k, int l, int r) {
    if (r <= a || b <= l)
      return def;
    if (a <= l && r <= b)
      return data[k];
    else {
      T c1 = _query(a, b, 2 * k + 1, l, (l + r) / 2);
      T c2 = _query(a, b, 2 * k + 2, (l + r) / 2, r);
      return operation(c1, c2);
    }
  }

public:
  SegTree(size_t _n, T _def, function<T(T, T)> _operation,
          function<T(T, T)> _update)
      : def(_def), operation(_operation), update(_update) {
    n = 1;
    while (n < _n) {
      n *= 2;
    }
    data = vector<T>(2 * n - 1, def);
  }

  void change(int i, T x) {
    i += n - 1;
    data[i] = update(data[i], x);
    while (i > 0) {
      i = (i - 1) / 2;
      data[i] = operation(data[i * 2 + 1], data[i * 2 + 2]);
    }
  }

  T query(int a, int b) { return _query(a, b, 0, 0, n); }

  T operator[](int i) { return data[i + n - 1]; }
};

vector<ll> divisor(ll n) {
  vector<ll> ret;
  for (ll i = 1; i * i <= n; i++) {
    if (n % i == 0) {
      ret.push_back(i);
      if (i * i != n)
        ret.push_back(n / i);
    }
  }
  sort(begin(ret), end(ret));
  return (ret);
}

/*
SegTree<ll> st(N, LLMAX,
        // a: left child, b: right child
        [](ll a, ll b) {return min(a, b); },
        // d: old, x: query
        [](ll d, ll x) {return x; });
// */

ll gcd(ll a, ll b) {
  if (a == 0)
    return b;
  if (b == 0)
    return a;

  if (a % b == 0) {
    return (b);
  } else {
    return (gcd(b, a % b));
  }
}

ll N, M, K, Q;
ll ans = 0;
ll H, W;

void solve() {
  // N = 100; Q = 100;
  cin >> H >> W >> K;

  ll xs, ys, xe, ye;
  cin >> xs >> ys >> xe >> ye;
  xs--;
  ys--;
  xe--;
  ye--;

  vvi G(H, vi(W, 0));
  rep(h, H) {
    rep(w, W) {
      char c;
      cin >> c;
      G[h][w] = (c == '.');
    }
  }

  priority_queue<piiii> que;
  vvi D(H, vi(W, LLMAX));
  D[xs][ys] = 0;
  que.push(mp(mp(0, -1), mp(xs, ys)));

  while (que.empty() == false) {
    piiii p = que.top();
    que.pop();
    ll d = -p.first.first;
    ll n = p.first.second;
    ll x = p.second.first;
    ll y = p.second.second;

    auto fn = [&](ll n, ll k, ll x, ll y) {
      ll nx = x;
      ll ny = y;

      if (n) {
        nx += k;
      } else {
        ny += k;
      }

      if (nx < 0 || H <= nx || ny < 0 || W <= ny)
        return false;
      if (!G[nx][ny])
        return false;

      ll ad = 1 + (abs(k) - 1) / K;
      if (D[nx][ny] < d + ad) {
        return false;
      }
      D[nx][ny] = d + ad;
      que.push(mp(mp(-d - ad, !n), mp(nx, ny)));
      return true;
    };

    if (n == -1 || n) {
      repa(k, 1, H) {
        if (!fn(true, k, x, y))
          break;
      }
      repa(k, 1, H) {
        if (!fn(true, -k, x, y))
          break;
      }
    }
    if (n == -1 || !n) {
      repa(k, 1, W) {
        if (!fn(false, k, x, y))
          break;
      }

      repa(k, 1, W) {
        if (!fn(false, -k, x, y))
          break;
      }
    }
  }
  if (D[xe][ye] == LLMAX) {
    cout << -1 << endl;
    ;
  } else {
    cout << D[xe][ye] << endl;
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  solve();
  return 0;
}