[Codetree] 되돌아오기 2 Python

https://www.codetree.ai/missions/5/problems/come-back-2/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

import sys


input  = sys.stdin.readline


x = 0
y = 0
d = 0


direct = {"N":0, "E": 1, "S":2, "W":3}

dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]

cnt = 0

cmd = list(input().rstrip())


for c in cmd:
    if c == 'L':
        d = (d+3)%4
    elif c == 'R':
        d = (d+1)%4
    else:
        x = x + dx[d]
        y = y + dy[d]
        
    cnt += 1

    if x == 0 and y == 0:
        print(cnt)
        exit()

print(-1)