ABC007 C - 幅優先探索

Calendar Clock iconCalendar Clock icon

atcoder

目次

# 問題

https://atcoder.jp/contests/abc007/tasks/abc007_3

# 入力

HWH W
sysxsy sx
gygxgy gx
c0,0c0,1...c0,W1c_{0,0} c_{0,1} ... c_{0,W-1}
...
cH1,0cH1,1...cH1,W1c_{H-1,0} c_{H-1,1} ... c_{H-1,W-1}.

  • HH - 迷路の高さ
  • WW - 迷路の幅
  • sysxsy sx - スタート地点の座標
  • gygxgy gx - ゴール地点の座標

# 出力

(sx,sy)(sx, sy)から(gx,gy)(gx, gy)に至る最短経路を出力する問題です.
上下左右に動くことが出来て、.は通ることが出来、#は通れません.

# 解説

最短経路と聞けばまっさきに幅優先探索が思い浮かびます.
幅優先探索では近い順に探索するので、最初にゴールを見つけた時に最短経路であることが保証されています.
すべてのマスを1回ずつ訪問するので計算量はO(HW)O(H W)です.

スタート地点からの距離の計算には動的計画法の考え方を使用します.
迷路と同じサイズの配列を用意し、\inftyで初期化します.
\inftyというのは、距離としてありえない範囲での最小の値です.
今回は迷路のサイズは50×5050 \times 50なので最大距離は約25002500です.
なので=3000\infty = 3000とします.
(大きすぎる値を設定しないのは計算時のオーバーフローを防ぐためです.)

スタート地点を距離00として、4近傍で未訪問(距離が\infty未満)の点に移動していきます.
距離は1つ移動するたびに前回いた点の距離に+1+1していきます.

訪問した点がゴールと等しければその点の距離を答えとします.

# 計算量

O(HW)O(H W)

# 解答

今回迷路と距離は2次元配列ではなく1次元配列で表現しています.
2次元配列の方が直感的で分かりやすいですが、vectorを使った一次元配列にも以下のようなメリットがあります.

    1. vector.atではC言語の配列のインデックスアクセスと違い不正なインデックスアクセスをした時のエラーが分かりやすい.
    1. サイズと初期値を指定した初期化が簡単.

WWの座標(x,y)(x, y)にはvector.at(yW+x)vector.at(y*W+x)でアクセス可能です.

// C++ 14
#include <iostream>
#include <string>
#include <vector>
#include <list>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <unordered_map>
#include <math.h>

#define ll long long
#define Int int
#define loop(x, start, end) for(Int x = start; x < end; x++)
#define loopdown(x, start, end) for(int x = start; x > end; x--)
#define rep(n) for(int x = 0; x < n; x++)
#define span(a,x,y) a.begin()+x,a.begin()+y
#define span_all(a) a.begin(),a.end()
#define len(x) (x.size())
#define last(x) (*(x.end()-1))

using namespace std;

#define EPS 0.0000000001
#define fequals(a,b) (fabs((a) - (b)) < EPS)

class Vector2 {
public:
  double x, y;
  
  Vector2(double x = 0, double y = 0): x(x), y(y) {}
  
  Vector2 operator + (const Vector2 v) const { return Vector2(x + v.x, y + v.y); }
  Vector2 operator - (const Vector2 v) const { return Vector2(x - v.x, y - v.y); }
  Vector2 operator * (const double k) const { return Vector2(x * k, y * k); }
  Vector2 operator / (const double k) const { return Vector2(x / k, y / k); }
  
  double length() { return sqrt(norm()); }
  double norm() { return x * x + y * y; }
  double dot (Vector2 const v) { return x * v.x + y * v.y; }
  double cross (Vector2 const v) { return x * v.y - y * v.x; }
  
  bool parallel(Vector2 &other) {
    return fequals(0, cross(other));
  }
  
  bool orthogonal(Vector2 &other) {
    return fequals(0, dot(other));
  }
  
  bool operator < (const Vector2 &v) {
    return x != v.x ? x < v.x : y < v.y;
  }
  
  bool operator == (const Vector2 &v) {
    return fabs(x - v.x) < EPS && fabs(y - v.y) < EPS;
  }
};

ostream & operator << (ostream & out, Vector2 const & v) { 
  out<< "Vector2(" << v.x << ", " << v.y << ')';
  return out;
}

istream & operator >> (istream & in, Vector2 & v) { 
  double x, y;
  in >> x;
  in >> y;
  v.x = x;
  v.y = y;
  return in;
}
#define MAX_N 50
#define INFTY 3000

Int H, W, sx, sy, gx, gy;
char c_;
vector<bool> M(MAX_N * MAX_N, true);
vector<Int> dist(MAX_N * MAX_N, INFTY);

void input() {
  cin >> H >> W >> sy >> sx >> gy >> gx;
  sx--, sy--, gx--, gy--;
  loop(h,0,H) {
    loop(w,0,W) {
      cin >> c_;
      if (c_ == '#') M[h*W + w] = false;
    }
  }
}

Int bfs(Int sx, Int sy) {
  queue<Vector2> Q;
  dist[sy*W + sx] = 0;
  Q.push(Vector2(sx, sy));

  while (Q.size()) {
    Vector2 u = Q.front(); Q.pop();
    loop(dx,-1,2) {
      loop(dy,-1,2) {
        if (dx != 0 && dy != 0) continue;
        Vector2 next = u + Vector2(dx, dy);
        if (next.x < 0 || next.x >= W || next.y < 0 || next.y >= H) continue;
        if (!M.at(next.y*W+next.x)) continue;
        if (dist.at(next.y*W + next.x) < INFTY) continue;
        dist[next.y*W+next.x] = dist[u.y*W+u.x] + 1;
        if (next.x == gx && next.y == gy) {
          return dist[next.y*W+next.x];
        }
        Q.push(next);
      }
    }
  }

  return INFTY;
}

void solve() {
  cout << bfs(sx, sy) << endl;
}

int main(void) {
  input();
  solve();
  return 0;
}

リモートフリーランス。ウェブサービス、スマホアプリエンジニア。
東アジアを拠点に世界を移動しながら活動してます!

お仕事のご依頼・お問い合わせはこちら

コメント