공원 산책

문제 설명

방향과 칸의 개수가 주어졌을 때, 경로에 장애물이 없거나 범위를 벗어나면 움직인다. 만일 장애물이 존재하거나 범위를 벗어난다면 해당 움직임은 철회한다. 이렇게 해서 마지막으로 도달하는 위치를 구하는 문제다.

https://school.programmers.co.kr/learn/courses/30/lessons/172928

프로그래머스

문제 풀이

첫 번째 접근

지금까지 풀어왓던 구현의 이동 문제와 크게 다르지 않다. 처음에는 시작 위치를 확인한다. 그리고 주어진 명령을 한 번씩 확인하는데, 북쪽이면 위로 올라가고 서쪽이면 왼쪽으로 움직이고 남쪽이면 아래로 내려가며 동쪽이면 오른쪽으로 간다. 이렇게 주어진 방향 만큼 한 칸씩 갈 때마다 해당 칸으로 갈 수 있는지 확인한다. 만일 갈 수 없다면 위치를 과거로 돌려주고 반복문을 탈출한다.

 

처음에는 주어진 칸 수만큼 움직이는 반복문을 일반화해보려고 했지만, 동서남북으로 움직일 때 주어지는 좌표의 방향이 다르기 때문에 그냥 하나씩 넣어줬다.

 

코드

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