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

#include <algorithm>
#include <array>
#include <assert.h>
#include <bitset>
#include <climits>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#define LEN(x) (long long)(x.size())
#define FOR(i, a, n) for (int i = (a); i < (n); ++i)
#define FOE(i, a) for (auto i : a)
#define ALL(c) (c).begin(), (c).end()
#define RALL(c) (c).rbegin(), (c).rend()
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v, x) (std::find(v.begin(), v.end(), x) != v.end())
#define BIT_COUNT(bit) (__builtin_popcount(bit))

typedef long long LL;
template <typename T> std::vector<T> make_v(size_t a) {
  return std::vector<T>(a);
}
template <typename T, typename... Ts> auto make_v(size_t a, Ts... ts) {
  return std::vector<decltype(make_v<T>(ts...))>(a, make_v<T>(ts...));
} // C++14
template <typename T, typename V>
typename std::enable_if<std::is_class<T>::value == 0>::type fill_v(T &t,
                                                                   const V &v) {
  t = v;
}
template <typename T, typename V>
typename std::enable_if<std::is_class<T>::value != 0>::type fill_v(T &t,
                                                                   const V &v) {
  for (auto &e : t)
    fill_v(e, v);
}
template <class T> inline T ceil(T a, T b) { return (a + b - 1) / b; }
void print() { std::cout << std::endl; }
template <class Head, class... Tail> void print(Head &&head, Tail &&...tail) {
  std::cout << head;
  if (sizeof...(tail) != 0) {
    std::cout << " ";
  }
  print(std::forward<Tail>(tail)...);
}
template <class T> void print(std::vector<T> &v) {
  for (auto &a : v) {
    std::cout << a;
    if (&a != &v.back()) {
      std::cout << " ";
    }
  }
  std::cout << std::endl;
}
template <class T> void print(std::vector<std::vector<T>> &vv) {
  for (auto &v : vv) {
    print(v);
  }
}
void debug() { std::cerr << std::endl; }
template <class Head, class... Tail> void debug(Head &&head, Tail &&...tail) {
  std::cerr << head;
  if (sizeof...(tail) != 0) {
    std::cerr << " ";
  }
  print(std::forward<Tail>(tail)...);
}
template <class T> void debug(std::vector<T> &v) {
  for (auto &a : v) {
    std::cerr << a;
    if (&a != &v.back()) {
      std::cerr << " ";
    }
  }
  std::cerr << std::endl;
}
template <class T> void debug(std::vector<std::vector<T>> &vv) {
  for (auto &v : vv) {
    print(v);
  }
}
inline bool inside(long long y, long long x, long long H, long long W) {
  return 0 <= y and y < H and 0 <= x and x < W;
}
template <class T> inline double euclidean_distance(T y1, T x1, T y2, T x2) {
  return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
template <class T> inline double manhattan_distance(T y1, T x1, T y2, T x2) {
  return abs(x1 - x2) + abs(y1 - y2);
}
template <typename T> T &chmin(T &a, const T &b) { return a = std::min(a, b); }
template <typename T> T &chmax(T &a, const T &b) { return a = std::max(a, b); }

// 初項s交差d長さnの数列の和
long long sum_of_arithmetic_progression(long long s, long long d, long long n) {
  return n * (2 * s + (n - 1) * d) / 2;
}

// xが2の階乗かどうか判定
bool is_power_of_two(long long x) { return !(x & (x - 1)); }

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

long long gcd(std::vector<long long> &v) {
  long long ans = v[0];
  for (int i = 1; i < (int)v.size(); ++i) {
    ans = gcd(ans, v[i]);
  }
  return ans;
}

long long lcm(long long a, long long b) {
  long long g = gcd(a, b);
  return a / g * b;
}

const int INF = 1u << 30u; // 1,073,741,824
const long long LINF = 1ull << 60u;
const double EPS = 1e-9;
const long double PI = acos(-1.0);
const std::vector<int> dy2 = {0, 1}, dx2 = {1, 0};
const std::vector<int> dy4 = {0, 1, 0, -1}, dx4 = {1, 0, -1, 0};
const std::vector<int> dy8 = {0, -1, 0, 1, 1, -1, -1, 1},
                       dx8 = {1, 0, -1, 0, 1, 1, -1, -1};

using namespace std;

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

  LL H, W, K;
  cin >> H >> W >> K;
  LL X1, Y1, X2, Y2;
  cin >> Y1 >> X1 >> Y2 >> X2;
  X1--;
  Y1--;
  X2--;
  Y2--;

  auto board = make_v<int>(H, W);
  FOR(y, 0, H) {
    FOR(x, 0, W) {
      char c;
      cin >> c;
      board[y][x] = (c == '@');
    }
  }

  auto cost = make_v<int>(H, W);
  fill_v(cost, INF);
  cost[Y1][X1] = 0;

  priority_queue<tuple<int, int, int>, vector<tuple<int, int, int>>,
                 greater<tuple<int, int, int>>>
      que;
  que.emplace(make_tuple(0, Y1, X1));

  while (not que.empty()) {
    int c, y, x;
    tie(c, y, x) = que.top();
    que.pop();

    FOR(i, 0, 4) {
      FOR(k, 1, K + 1) {
        int ny = dy4[i] * k + y;
        int nx = dx4[i] * k + x;
        if (inside(ny, nx, H, W) and cost[ny][nx] >= c + 1 and
            board[ny][nx] == 0) {
          if (cost[ny][nx] == c + 1) {
            continue;
          }
          que.emplace(make_tuple(c + 1, ny, nx));
          cost[ny][nx] = c + 1;
        } else {
          break;
        }
      }
    }
  }

  if (cost[Y2][X2] == INF) {
    print(-1);
  } else {
    print(cost[Y2][X2]);
  }

  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <array>
#include <assert.h>
#include <bitset>
#include <climits>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#define LEN(x) (long long)(x.size())
#define FOR(i, a, n) for (int i = (a); i < (n); ++i)
#define FOE(i, a) for (auto i : a)
#define ALL(c) (c).begin(), (c).end()
#define RALL(c) (c).rbegin(), (c).rend()
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v, x) (std::find(v.begin(), v.end(), x) != v.end())
#define BIT_COUNT(bit) (__builtin_popcount(bit))

typedef long long LL;
template <typename T> std::vector<T> make_v(size_t a) {
  return std::vector<T>(a);
}
template <typename T, typename... Ts> auto make_v(size_t a, Ts... ts) {
  return std::vector<decltype(make_v<T>(ts...))>(a, make_v<T>(ts...));
} // C++14
template <typename T, typename V>
typename std::enable_if<std::is_class<T>::value == 0>::type fill_v(T &t,
                                                                   const V &v) {
  t = v;
}
template <typename T, typename V>
typename std::enable_if<std::is_class<T>::value != 0>::type fill_v(T &t,
                                                                   const V &v) {
  for (auto &e : t)
    fill_v(e, v);
}
template <class T> inline T ceil(T a, T b) { return (a + b - 1) / b; }
void print() { std::cout << std::endl; }
template <class Head, class... Tail> void print(Head &&head, Tail &&...tail) {
  std::cout << head;
  if (sizeof...(tail) != 0) {
    std::cout << " ";
  }
  print(std::forward<Tail>(tail)...);
}
template <class T> void print(std::vector<T> &v) {
  for (auto &a : v) {
    std::cout << a;
    if (&a != &v.back()) {
      std::cout << " ";
    }
  }
  std::cout << std::endl;
}
template <class T> void print(std::vector<std::vector<T>> &vv) {
  for (auto &v : vv) {
    print(v);
  }
}
void debug() { std::cerr << std::endl; }
template <class Head, class... Tail> void debug(Head &&head, Tail &&...tail) {
  std::cerr << head;
  if (sizeof...(tail) != 0) {
    std::cerr << " ";
  }
  print(std::forward<Tail>(tail)...);
}
template <class T> void debug(std::vector<T> &v) {
  for (auto &a : v) {
    std::cerr << a;
    if (&a != &v.back()) {
      std::cerr << " ";
    }
  }
  std::cerr << std::endl;
}
template <class T> void debug(std::vector<std::vector<T>> &vv) {
  for (auto &v : vv) {
    print(v);
  }
}
inline bool inside(long long y, long long x, long long H, long long W) {
  return 0 <= y and y < H and 0 <= x and x < W;
}
template <class T> inline double euclidean_distance(T y1, T x1, T y2, T x2) {
  return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
template <class T> inline double manhattan_distance(T y1, T x1, T y2, T x2) {
  return abs(x1 - x2) + abs(y1 - y2);
}
template <typename T> T &chmin(T &a, const T &b) { return a = std::min(a, b); }
template <typename T> T &chmax(T &a, const T &b) { return a = std::max(a, b); }

// 初項s交差d長さnの数列の和
long long sum_of_arithmetic_progression(long long s, long long d, long long n) {
  return n * (2 * s + (n - 1) * d) / 2;
}

// xが2の階乗かどうか判定
bool is_power_of_two(long long x) { return !(x & (x - 1)); }

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

long long gcd(std::vector<long long> &v) {
  long long ans = v[0];
  for (int i = 1; i < (int)v.size(); ++i) {
    ans = gcd(ans, v[i]);
  }
  return ans;
}

long long lcm(long long a, long long b) {
  long long g = gcd(a, b);
  return a / g * b;
}

const int INF = 1u << 30u; // 1,073,741,824
const long long LINF = 1ull << 60u;
const double EPS = 1e-9;
const long double PI = acos(-1.0);
const std::vector<int> dy2 = {0, 1}, dx2 = {1, 0};
const std::vector<int> dy4 = {0, 1, 0, -1}, dx4 = {1, 0, -1, 0};
const std::vector<int> dy8 = {0, -1, 0, 1, 1, -1, -1, 1},
                       dx8 = {1, 0, -1, 0, 1, 1, -1, -1};

using namespace std;

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

  LL H, W, K;
  cin >> H >> W >> K;
  LL X1, Y1, X2, Y2;
  cin >> Y1 >> X1 >> Y2 >> X2;
  X1--;
  Y1--;
  X2--;
  Y2--;

  auto board = make_v<int>(H, W);
  FOR(y, 0, H) {
    FOR(x, 0, W) {
      char c;
      cin >> c;
      board[y][x] = (c == '@');
    }
  }

  auto cost = make_v<int>(H, W);
  fill_v(cost, INF);

  priority_queue<tuple<int, int, int>, vector<tuple<int, int, int>>,
                 greater<tuple<int, int, int>>>
      que;
  que.emplace(make_tuple(0, Y1, X1));

  while (not que.empty()) {
    int c, y, x;
    tie(c, y, x) = que.top();
    que.pop();

    if (cost[y][x] <= c) {
      continue;
    }
    cost[y][x] = c;

    FOR(i, 0, 4) {
      FOR(k, 1, K + 1) {
        int ny = dy4[i] * k + y;
        int nx = dx4[i] * k + x;
        if (inside(ny, nx, H, W) and cost[ny][nx] == INF and
            board[ny][nx] == 0) {
          que.emplace(make_tuple(c + 1, ny, nx));
        } else {
          break;
        }
      }
    }
  }

  if (cost[Y2][X2] == INF) {
    print(-1);
  } else {
    print(cost[Y2][X2]);
  }

  return 0;
}