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

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)
using namespace std;
using ll = long long;
using P = pair<int, int>;

const int di[] = {-1, 0, 1, 0};
const int dj[] = {0, -1, 0, 1};

int main() {
  int h, w;
  cin >> h >> w;
  int si, sj;
  cin >> si >> sj;
  int ti, tj;
  cin >> ti >> tj;
  --si;
  --sj;
  --ti;
  --tj;
  vector<string> s(h);
  rep(i, h) cin >> s[i];
  const int INF = 1e9;
  vector<vector<int>> dist(h, vector<int>(w, INF));
  deque<P> q;
  dist[si][sj] = 0;
  q.emplace_back(si, sj);
  while (!q.empty()) {
    int i = q.front().first;
    int j = q.front().second;
    int d = dist[i][j];
    q.pop_front();
    if (dist[i][j] != d)
      continue;
    rep(v, 4) {
      int ni = i + di[v], nj = j + dj[v];
      if (ni < 0 || ni >= h || nj < 0 || nj >= w)
        continue;
      if (s[ni][nj] == '#')
        continue;
      if (dist[ni][nj] <= d)
        continue;
      dist[ni][nj] = d;
      q.emplace_front(ni, nj);
    }
    for (int ei = -2; ei <= 2; ++ei) {
      for (int ej = -2; ej <= 2; ++ej) {
        int ni = i + ei, nj = j + ej;
        if (ni < 0 || ni >= h || nj < 0 || nj >= w)
          continue;
        if (s[ni][nj] == '#')
          continue;
        if (dist[ni][nj] <= d + 1)
          continue;
        dist[ni][nj] = d + 1;
        q.emplace_back(ni, nj);
      }
    }
  }
  int ans = dist[ti][tj];
  if (ans == INF)
    ans = -1;
  cout << ans << endl;
  return 0;
}

Error: TLE

Fixed submission:

#include <algorithm>
#include <cmath>
#include <deque>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <vector>

#define ll long long
#define REP(i, n) for (int i = 0; i < n; i++)
#define PI 3.141592653589
#define MOD 1000000007

using namespace std;

template <typename T> T gcd(T a, T b) { return (a % b) ? gcd(b, a % b) : b; }
template <typename T> T lcm(T a, T b) { return a * b / gcd(a, b); }

const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, 1, 0, -1};

int main(void) {
  /*
  BFSで解くよ
  スタートからいけるところに「0」を書く
  いけなくなったら，「0」からいけるところに「1」を書く
  01-BFS　コストが0と1しかないので　それ以外ならダイクストラを使う
  01-BFSを実現するためには，deque（デック）を使うといい　ダイクストラはデータのプッシュにO(log
  N)かかっちゃうのでdequeで
  */

  int h, w;
  cin >> h >> w;
  int ch, cw, dh, dw;
  cin >> ch >> cw;
  cin >> dh >> dw;
  ch--;
  cw--;
  dh--;
  dw--;

  vector<string> s(h);
  REP(i, h) cin >> s[i];

  const int INF = 1e9;
  vector<vector<int>> dist(h, vector<int>(w, INF));

  deque<pair<int, int>> q; // dequeに座標が入る
  dist[ch][cw] = 0;
  // pair<int, int> p = make_pair(ch, cw);
  q.emplace_back(ch, cw); // emplace_backはほとんどpush_back
  // q.push_back(p);

  while (!q.empty()) {
    // 前から取り出す
    int i = q.front().first;
    int j = q.front().second;
    q.pop_front();
    int d = dist[i][j];

    // コスト0の移動
    REP(dir, 4) {
      int newi = i + dx[dir], newj = j + dy[dir];
      if (newi < 0 || newi >= h || newj < 0 || newj >= w)
        continue;
      if (s[newi][newj] == '#')
        continue;

      // そもそも距離が更新できないなら遷移する意味ない
      if (dist[newi][newj] <= d)
        continue;

      dist[newi][newj] = d;
      // pair<int, int> p = make_pair(newi, newj);
      q.emplace_front(newi, newj);
      // q.push_front(p);
    }

    // コスト1の移動
    for (int ei = -2; ei <= 2; ei++) {
      for (int ej = -2; ej <= 2; ej++) {
        int newi = i + ei, newj = j + ej;

        if (newi < 0 || newi >= h || newj < 0 || newj >= w)
          continue;
        if (s[newi][newj] == '#')
          continue;

        // そもそも距離が更新できないなら遷移する意味ない
        if (dist[newi][newj] <= d + 1)
          continue;

        dist[newi][newj] = d + 1;
        // pair<int, int> p = make_pair(newi, newj);
        q.emplace_front(newi, newj);
        // q.push_front(p);
      }
    }
  }

  int ans = dist[dh][dw];
  if (ans == INF)
    ans = -1;
  cout << ans << endl;

  return 0;
}