algorithm/코딩 테스트
공원 산책
토끼띠NJW
2023. 6. 18. 23:12
문제 설명
방향과 칸의 개수가 주어졌을 때, 경로에 장애물이 없거나 범위를 벗어나면 움직인다. 만일 장애물이 존재하거나 범위를 벗어난다면 해당 움직임은 철회한다. 이렇게 해서 마지막으로 도달하는 위치를 구하는 문제다.
https://school.programmers.co.kr/learn/courses/30/lessons/172928
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제 풀이
첫 번째 접근
지금까지 풀어왓던 구현의 이동 문제와 크게 다르지 않다. 처음에는 시작 위치를 확인한다. 그리고 주어진 명령을 한 번씩 확인하는데, 북쪽이면 위로 올라가고 서쪽이면 왼쪽으로 움직이고 남쪽이면 아래로 내려가며 동쪽이면 오른쪽으로 간다. 이렇게 주어진 방향 만큼 한 칸씩 갈 때마다 해당 칸으로 갈 수 있는지 확인한다. 만일 갈 수 없다면 위치를 과거로 돌려주고 반복문을 탈출한다.
처음에는 주어진 칸 수만큼 움직이는 반복문을 일반화해보려고 했지만, 동서남북으로 움직일 때 주어지는 좌표의 방향이 다르기 때문에 그냥 하나씩 넣어줬다.
코드
def solution(park, routes):
answer = []
n = len(park)
m = len(park[0])
r = 0
c = 0
for i in range(n):
for j in range(m):
if park[i][j] == 'S':
r = i
c = j
for route in routes:
d = route[0]
count = int(route[2])
tmpR = r
tmpC = c
if d == 'N':
for i in range(count):
r -= 1
if 0 <= r < n and 0 <= c < m:
if park[r][c] == 'O' or park[r][c] == 'S':
continue
else:
r = tmpR
c = tmpC
break
else:
r = tmpR
c = tmpC
break
elif d == 'S':
for j in range(count):
r += 1
if 0 <= r < n and 0 <= c < m:
if park[r][c] == 'O' or park[r][c] == 'S':
continue
else:
r = tmpR
c = tmpC
break
else:
r = tmpR
c = tmpC
break
elif d == 'W':
for j in range(count):
c -= 1
if 0 <= r < n and 0 <= c < m:
if park[r][c] == 'O' or park[r][c] == 'S':
continue
else:
r = tmpR
c = tmpC
break
else:
r = tmpR
c = tmpC
break
elif d == 'E':
for j in range(count):
c += 1
if 0 <= r < n and 0 <= c < m:
if park[r][c] == 'O' or park[r][c] == 'S':
continue
else:
r = tmpR
c = tmpC
break
else:
r = tmpR
c = tmpC
break
answer.append(r)
answer.append(c)
return answer