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

/**
 *    author:   K.R.Surrya
 *    created:  15.06.2020 17:06:33
 **/
//~ ~~ ~~~~ ~~~~~~~~
#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
using pint = pair<int, int>;
using plint = pair<lint, lint>;
#define vi vector<int>
#define vli vector<lint>
#define pb push_back
#define ALL(x) begin(x), end(x)
#define inf (1LL < 63) - 1
#define MOD 1000000007
#define SZ(x) ((lint)(x).size())
#define FOR(i, begin, end)                                                     \\
  for (int i = (begin), i##_end_ = (end); i < i##_end_; i++)
#define IFOR(i, begin, end)                                                    \\
  for (int i = (end)-1, i##_begin_ = (begin); i >= i##_begin_; i--)
#define REP(i, n) FOR(i, 0, n)
#define IREP(i, n) IFOR(i, 0, n)
#define f_ip                                                                   \\
  ios_base::sync_with_stdio(false);                                            \\
  cin.tie(NULL);
#define ZERO(a) fill(ALL(a), 0)
struct fast_ios {
  fast_ios() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(20);
  };
} fast_ios_;
template <typename T> istream &operator>>(istream &is, vector<T> &vec) {
  for (auto &v : vec)
    is >> v;
  return is;
}
template <typename T> bool chmax(T &m, const T q) {
  if (m < q) {
    m = q;
    return true;
  } else
    return false;
}
template <typename T> bool chmin(T &m, const T q) {
  if (q < m) {
    m = q;
    return true;
  } else
    return false;
}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

/* ~Debugging~ */
/*~~~~~~~~~~~~~~~~~*/
#define LOCAL
#define sim template <class c
#define ris return *this
#define dor > deb &operator<<
#define eni(x)                                                                 \\
  sim > typename enable_if<sizeof dud<c>(0) x 1, deb &>::type operator<<(c i) {
sim > struct rge {
  c b, e;
};
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c *x) -> decltype(cerr << *x, 0);
sim > char dud(...);
struct deb {
#ifdef LOCAL
  ~deb() { cerr << endl; }
  eni(!=) cerr << boolalpha << i;
  ris;
} eni(==) ris << range(begin(i), end(i));
}
sim, class b dor(pair<b, c> d) {
  ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
  *this << "[";
  for (auto it = d.b; it != d.e; ++it)
    *this << ", " + 2 * (it == d.b) << *it;
  ris << "]";
}
#else
  sim dor(const c &) { ris; }
#endif
}
;
#define what(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
// -DLOCAL

/*  ~magic~ */
/*~~~~~~~~~~~~~~~*/
#define magic(args...)                                                         \\
  {                                                                            \\
    string _s = #args;                                                         \\
    replace(_s.begin(), _s.end(), ',', ' ');                                   \\
    stringstream _ss(_s);                                                      \\
    istream_iterator<string> _it(_ss);                                         \\
    err(_it, args);                                                            \\
  }
void err(istream_iterator<string> it) { return; }
template <typename T, typename... Args>
void err(istream_iterator<string> it, T a, Args... args) {
  cerr << *it << " = " << a << endl;
  err(++it, args...); // args is going to end.
}

/* ~driver~ */
/*~~~~~~~~~~*/
vector<int> dx = {0, 1, 0, -1}; // clockwise from east direction;
vector<int> dy = {1, 0, -1, 0};
struct mPair {

  int x;
  int y;
  string path;

  mPair() {}

  mPair(int x, int y, string path) {
    this->x = x;
    this->y = y;
    this->path = path;
  }
};

bool isv(int x, int y, int n, int m) {
  if (x >= 0 && x <= n - 1 && y >= 0 && y <= m - 1)
    return true;
  else
    return false;
}

void solve() {
  int H, W, cap;
  cin >> H >> W >> cap;
  int sx, sy;
  cin >> sx >> sy;
  --sx;
  --sy;
  int destx, desty;
  cin >> destx >> desty;
  --destx;
  --desty;
  vector<vector<int>> mat(H, vector<int>(W));
  REP(i, H) {
    REP(j, W) {
      char ch;
      cin >> ch;       // only two types of characters present in the matrix.
      if (ch == '@') { // acts as a blocker;
        mat[i][j] = 0;
      } else {
        mat[i][j] = 1;
      }
    }
  }

  vector<vector<lint>> dist(H, vector<lint>(W, 1e18));
  queue<mPair> q;
  q.push({sx, sy, ""});
  bool ok = 0;
  string check;
  dist[sx][sy] = 0;
  while (q.size()) {
    mPair rem = q.front();
    q.pop();
    if (rem.x == destx && rem.y == desty) {
      cout << dist[rem.x][rem.y] << '\\n';
      exit(0);
      check = rem.path;
      ok = 1;
      break;
    }
    REP(k, 4) {
      FOR(leap, 1, cap + 1) {
        int nx = rem.x + dx[k] * leap;
        int ny = rem.y + dy[k] * leap;
        if (!isv(nx, ny, H, W))
          continue; // validity
        if (mat[nx][ny] == 0 || dist[nx][ny] <= dist[rem.x][rem.y])
          break; // road_block  //stop redundant work
        /* important */
        if (dist[nx][ny] > dist[rem.x][rem.y] + 1) {
          string curr = rem.path + to_string(k);
          dist[nx][ny] = dist[rem.x][rem.y] + 1;
          q.push({nx, ny, curr});
        }
      }
    }
  }

  cout << -1 << '\\n';
}

int main() {
#if 0
     freopen("input.txt", "r", stdin);
     freopen("output.txt", "w", stdout);
#endif

  int T = 1;
  // cin>>T;
  for (int tt = 1; tt <= T; ++tt) {
    solve();
  }

  return 0;
}

/* ~Modulus functions~ */
/*~~~~~~~~~~~~~~~~~~~~~~~*/
lint pow_mod(lint base, lint expo) {
  lint ret = 1;
  for (; expo;) {
    if (expo & 1)
      ret = (ret * base) % MOD;

    expo = (expo >> 1);
    base = (base * base) % MOD;
  }
  return ret;
}


Error: TLE

Fixed submission:

/**
 *    author:   K.R.Surrya
 *    created:  15.06.2020 17:06:33
 **/
//~ ~~ ~~~~ ~~~~~~~~
#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
using pint = pair<int, int>;
using plint = pair<lint, lint>;
#define vi vector<int>
#define vli vector<lint>
#define pb push_back
#define ALL(x) begin(x), end(x)
#define inf (1LL < 63) - 1
#define MOD 1000000007
#define SZ(x) ((lint)(x).size())
#define FOR(i, begin, end)                                                     \\
  for (int i = (begin), i##_end_ = (end); i < i##_end_; i++)
#define IFOR(i, begin, end)                                                    \\
  for (int i = (end)-1, i##_begin_ = (begin); i >= i##_begin_; i--)
#define REP(i, n) FOR(i, 0, n)
#define IREP(i, n) IFOR(i, 0, n)
#define f_ip                                                                   \\
  ios_base::sync_with_stdio(false);                                            \\
  cin.tie(NULL);
#define ZERO(a) fill(ALL(a), 0)
struct fast_ios {
  fast_ios() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(20);
  };
} fast_ios_;
template <typename T> istream &operator>>(istream &is, vector<T> &vec) {
  for (auto &v : vec)
    is >> v;
  return is;
}
template <typename T> bool chmax(T &m, const T q) {
  if (m < q) {
    m = q;
    return true;
  } else
    return false;
}
template <typename T> bool chmin(T &m, const T q) {
  if (q < m) {
    m = q;
    return true;
  } else
    return false;
}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

/* ~Debugging~ */
/*~~~~~~~~~~~~~~~~~*/
#define LOCAL
#define sim template <class c
#define ris return *this
#define dor > deb &operator<<
#define eni(x)                                                                 \\
  sim > typename enable_if<sizeof dud<c>(0) x 1, deb &>::type operator<<(c i) {
sim > struct rge {
  c b, e;
};
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c *x) -> decltype(cerr << *x, 0);
sim > char dud(...);
struct deb {
#ifdef LOCAL
  ~deb() { cerr << endl; }
  eni(!=) cerr << boolalpha << i;
  ris;
} eni(==) ris << range(begin(i), end(i));
}
sim, class b dor(pair<b, c> d) {
  ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
  *this << "[";
  for (auto it = d.b; it != d.e; ++it)
    *this << ", " + 2 * (it == d.b) << *it;
  ris << "]";
}
#else
  sim dor(const c &) { ris; }
#endif
}
;
#define what(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
// -DLOCAL

/*  ~magic~ */
/*~~~~~~~~~~~~~~~*/
#define magic(args...)                                                         \\
  {                                                                            \\
    string _s = #args;                                                         \\
    replace(_s.begin(), _s.end(), ',', ' ');                                   \\
    stringstream _ss(_s);                                                      \\
    istream_iterator<string> _it(_ss);                                         \\
    err(_it, args);                                                            \\
  }
void err(istream_iterator<string> it) { return; }
template <typename T, typename... Args>
void err(istream_iterator<string> it, T a, Args... args) {
  cerr << *it << " = " << a << endl;
  err(++it, args...); // args is going to end.
}

/* ~driver~ */
/*~~~~~~~~~~*/
vector<int> dx = {0, 1, 0, -1}; // clockwise from east direction;
vector<int> dy = {1, 0, -1, 0};
struct mPair {

  int x;
  int y;
  string path;

  mPair() {}

  mPair(int x, int y, string path) {
    this->x = x;
    this->y = y;
    this->path = path;
  }
};

bool isv(int x, int y, int n, int m) {
  if (x >= 0 && x <= n - 1 && y >= 0 && y <= m - 1)
    return true;
  else
    return false;
}

void solve() {
  int H, W, cap;
  cin >> H >> W >> cap;
  int sx, sy;
  cin >> sx >> sy;
  --sx;
  --sy;
  int destx, desty;
  cin >> destx >> desty;
  --destx;
  --desty;
  vector<vector<int>> mat(H, vector<int>(W));
  REP(i, H) {
    REP(j, W) {
      char ch;
      cin >> ch;       // only two types of characters present in the matrix.
      if (ch == '@') { // acts as a blocker;
        mat[i][j] = 0;
      } else {
        mat[i][j] = 1;
      }
    }
  }

  vector<vector<lint>> dist(H, vector<lint>(W, 1e18));
  queue<mPair> q;
  q.push({sx, sy, ""});
  bool ok = 0;
  string check;
  dist[sx][sy] = 0;
  while (q.size()) {
    mPair rem = q.front();
    q.pop();
    if (rem.x == destx && rem.y == desty) {
      cout << dist[rem.x][rem.y] << '\\n';
      exit(0);
      check = rem.path;
      ok = 1;
      break;
    }
    REP(k, 4) {
      FOR(leap, 1, cap + 1) {
        int nx = rem.x + dx[k] * leap;
        int ny = rem.y + dy[k] * leap;
        if (!isv(nx, ny, H, W))
          continue;
        if (mat[nx][ny] == 0)
          break;
        /* important */
        if (dist[nx][ny] > dist[rem.x][rem.y] + 1) {
          string curr = rem.path + to_string(k);
          dist[nx][ny] = dist[rem.x][rem.y] + 1;
          q.push({nx, ny, curr});
        }
      }
    }
  }

  cout << -1 << '\\n';
}

int main() {
#if 0
     freopen("input.txt", "r", stdin);
     freopen("output.txt", "w", stdout);
#endif

  int T = 1;
  // cin>>T;
  for (int tt = 1; tt <= T; ++tt) {
    solve();
  }

  return 0;
}

/* ~Modulus functions~ */
/*~~~~~~~~~~~~~~~~~~~~~~~*/
lint pow_mod(lint base, lint expo) {
  lint ret = 1;
  for (; expo;) {
    if (expo & 1)
      ret = (ret * base) % MOD;

    expo = (expo >> 1);
    base = (base * base) % MOD;
  }
  return ret;
}