h_nosonの日記

競プロ、CTFなど

yukicoder No.367 ナイトの転身

問題
No.367 ナイトの転身 - yukicoder
H\times Wのチェス盤が与えられ,ナイトとミニビショップを操作する.
ミニビショップは斜めに1マスだけ移動できる駒である.
ナイトでスタートし,赤いマスを通るとミニビショップになる.
スタート地点,ゴール地点,赤いマスが与えられた時,
ゴールするための最小の移動回数はいくつか.

解法
幅優先探索を使う.
ナイト用とミニビショップ用の2つの盤の配列を用意する点を除いて,基本的な幅優先探索と変わらない.
ナイトの移動は

int dxy = {2,1,2,-1,-2,1,-2,-1};
for (i = 0; i < 8; i++) {
    int nx = x + dxy[i];
    int ny = y + dxy[(i+1)%8];
    // code
}

とすれば綺麗に書ける.

ソースコード

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;

#define RREP(i,s,e) for (i = s; i >= e; i--)
#define rrep(i,n) RREP(i,n-1,0)
#define REP(i,s,e) for (i = s; i < e; i++)
#define rep(i,n) REP(i,0,n)
#define INF 100000000

typedef long long ll;

int main() {
    int i, j, k, l, h, w;
    int dis[2][500][500];
    string s[500];
    cin >> h >> w;
    rep (i,h) cin >> s[i];
    rep (i,2) rep (j,h) rep (k,w) dis[i][j][k] = INF;
    rep (i,h) rep (j,w) if (s[i][j] == 'S') {
        queue<tuple<int,int,int>> q;
        int kn[8] = {2,1,2,-1,-2,1,-2,-1};
        dis[1][i][j] = 0;
        q.push(make_tuple(i,j,true));
        while (!q.empty()) {
            int x, y, d;
            bool knight;
            tie(x,y,knight) = q.front();
            q.pop();
            if (s[x][y] == 'G') {
                cout << dis[knight][x][y] << endl;
                return 0;
            }
            d = dis[knight][x][y];
            knight ^= s[x][y] == 'R';
            if (knight) {
                rep (k,8) {
                    int nx = x+kn[k];
                    int ny = y+kn[(k+1)%8];
                    if (nx >= 0 && nx < h && ny >= 0 && ny < w && dis[knight][nx][ny] > d+1) {
                        dis[knight][nx][ny] = d+1;
                        q.push(make_tuple(nx,ny,knight));
                    }
                }
            }
            else {
                for (k = -1; k <= 1; k+=2) {
                    for (l = -1; l <= 1; l+=2) {
                        int nx = x+k;
                        int ny = y+l;
                        if (nx >= 0 && nx < h && ny >= 0 && ny < w && dis[knight][nx][ny] > d+1) {
                            dis[knight][nx][ny] = d+1;
                            q.push(make_tuple(nx,ny,knight));
                        }
                    }
                }
            }
        }
    }
    cout << -1 << endl;
    return 0;
}