paizaラーニング問題集「移動が可能かの判定・複数回の移動」を解いてみた

▼考え方

この問題を解くために私が考えた内容1～3を以下に示します。

1. 現在向いている方角を示すneswと移動の向きdを関数caldistanceに渡して、y,x座標の移動距離を計算します。

2. 現在の座標(y,x)と1の戻り値(y,x座標の移動距離)を関数nextpositioncheckに渡して、移動が可能かどうかを判定します。題意の条件に合致した場合は"Stop"を出力します。

3. 移動後の向いている方角と移動後の座標を求めます。

▼コード

H,W,sy,sx,N = map(int,input().split())

S = [list(input()) for _ in range(H)]
d = [input() for _ in range(N)]

# nesw: 現在向いている方角を示す変数 0:北,1:東,2:南,3:西
nesw = 0

# y: 現在のy座標を示す変数
y = sy

# x: 現在のx座標を示す変数
x = sx

# caldistance: y,x座標の移動距離を計算を計算する関数
def caldistance(nesw,d_i):

    # dy: y座標の移動距離を示す変数
    dy = 0

    # dx: x座標の移動距離を示す変数
    dx = 0

    if d_i == "L":
    
        if nesw == 0:
            dx -= 1
        elif nesw == 1:
            dy -= 1
        elif nesw == 2:
            dx += 1
        else:
            dy += 1

    else:
    
        if nesw == 0:
            dx += 1
        elif nesw == 1:
            dy += 1
        elif nesw == 2:
            dx -= 1
        else:
            dy -= 1

    return [dy,dx]

# nextpositioncheck: 移動が可能かどうかを判定する関数
def nextpositioncheck(y,x,dyx):
    
    if (y+dyx[0] <= -1 or y+dyx[0] >= H) or (x+dyx[1] <= -1  or x+dyx[1] >= W):
        print("Stop")
        exit()

    if S[y+dyx[0]][x+dyx[1]] == "#":
        print("Stop")
        exit()

for i in range(N):

    # 考え方1.
    # dyx: y,x座標の移動距離に関する情報を格納するリスト
    dyx = caldistance(nesw,d[i])
    
    # 考え方2.
    nextpositioncheck(y,x,dyx)

    # 考え方3.
    if d[i] == "L":
        nesw -= 1
    else:
        nesw += 1
    
    if nesw > 3:
        nesw = 0
    elif nesw < 0:
        nesw = 3
    
    y += dyx[0]
    x += dyx[1]
    print(y,x)