🤖 옌지니어의 IT 기록 🤖

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

문제

문제 설명

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한 사항

• 3 ≤ park의 길이 ≤ 50
  • 3 ≤ park[i]의 길이 ≤ 50
    • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
      • S : 시작 지점
      • O : 이동 가능한 통로
      • X : 장애물
  • park는 직사각형 모양입니다.
• 1 ≤ routes의 길이 ≤ 50
  • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
  • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
  • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
    • op는 다음 네 가지중 하나로 이루어져 있습니다.
      • N : 북쪽으로 주어진 칸만큼 이동합니다.
      • S : 남쪽으로 주어진 칸만큼 이동합니다.
      • W : 서쪽으로 주어진 칸만큼 이동합니다.
      • E : 동쪽으로 주어진 칸만큼 이동합니다.
    • 1 ≤ n ≤ 9

입출력 예

소스 코드

# 1
def solution(park, routes):
    # x = 열 번호 / y = 행 번호
    # 행열의 최대값 설정
    x_max = len(park[0])
    y_max = len(park)
    
    # 시작 위치 찾기
    x = 0
    y = 0
    for i in range(y_max):          # i = 행 번호 = y값
        for j in range(x_max):      # j = 열 번호 = x값
            if park[i][j] == 'S':
                x = j
                y = i
                break
    
    # 좌표 이동
    for route in routes:
        # 방향과 이동거리를 저장
        direction, distance = route.split(" ")

        # 현재의 좌표를 임시로 이동
        xx = x
        yy = y

        # 한 칸씩 이동하면서 공원을 벗어나거나 장애물을 만나는지 확인
        # step이 이동거리와 같아지면 현재 좌표를 임시 좌표로 변경
        for step in range(1, int(distance)+1):
            # 동쪽으로 이동
            if direction == 'E' and xx != x_max-1 and park[yy][xx+1] != 'X':
                xx += 1
                if step == int(distance):
                    x = xx
            # 서쪽으로 이동
            elif direction == 'W' and xx != 0 and park[yy][xx-1] != 'X':
                xx -= 1
                if step == int(distance):
                    x = xx
            # 남쪽으로 이동
            elif direction == 'S' and yy != y_max-1 and park[yy+1][xx] != 'X':
                yy += 1
                if step == int(distance):
                    y = yy
            # 북쪽으로 이동
            elif direction == 'N' and yy != 0 and park[yy-1][xx] != 'X':
                yy -= 1
                if step == int(distance):
                    y = yy

    return [y, x]

풀이 # 1

• 좌표에서 고려사항
  • x = 열 번호 / y = 행 번호
  • x_max = 열의 길이 / y_max = 행의 길이
• 시작 위치 찾기
  • i = 행 번호 = y값 / j = 열 번호 / x값
  • 위의 내용을 고려하여 시작점 S 찾기
• 좌표 이동
  • 띄어쓰기를 기준으로 앞의 값은 방향(direction), 뒤의 값은 이동거리(distance)
  • 현재 좌표와 동일한 임시 좌표를 만들어서 이동 (조건을 만족시키지 않을 경우 초기화하기 위해 필요)
    • 방향 확인
      • E = 동쪽 : x좌표 + 1
      • W = 서쪽 : x좌표 - 1
      • S = 남쪽 : y좌표 + 1
      • N = 북쪽 : y좌표 -1
    • 현재 좌표가 행이나 열의 길이를 벗어나는지 확인
    • 한 칸 이동했을 때 장애물이 있는지 확인
  • 각 조건을 모두 만족한 경우 임시 좌표를 한 칸 이동
  • step이 이동 거리에 도달하면 현재 좌표를 임시 좌표로 바꾸기
  • x는 열 번호, y는 행 번호이므로 [y,x] 를 반환

# 2
def solution(park, routes):
    # 행열의 최대값 설정
    w, h = len(park[0]), len(park)

    # 방향별로 이동해야하는 거리를 딕셔너리에 저장
    move = {"N":(-1,0), "S":(1,0), "W":(0,-1), "E":(0,1)}
    
    # 시작 위치 찾기
    x, y = 0, 0
    for i in range(h):
        for j in range(w):
            if park[i][j] == "S":
                x, y = i, j
                break
        
    # 좌표 이동
    for route in routes:
        # 방향과 이동거리를 저장
        direction, distance = route.split(" ")

        # 현재의 좌표를 저장해서 조건을 만족하지 못하면 해당값으로 초기화
        dx, dy = x, y
        
        for idx in range(int(distance)):
            # 이동할 위치
            nx = x + move[direction][0]
            ny = y + move[direction][1]
        
            if 0 <= nx <= h-1 and 0 <= ny <= w-1 and (park[nx][ny]!="X"):
                x, y = nx, ny                    
            else:
                x, y = dx, dy
                break

    return [x,y]

풀이 #2

• 위의 풀이와 반대로 x를 행 번호, y를 행 번호로 생각한 풀이
• 키가 이동 방향, 값이 이동해야하는 거리인 딕셔너리 생성
  • E = 동쪽 : y좌표 + 1
  • W = 서쪽 : y좌표 - 1
  • S = 남쪽 : x좌표 + 1
  • N = 북쪽 : x좌표 -1
• 시작 위치 찾기
• 현재 좌표를 dx, dy에 저장 (조건을 만족시키지 않을 경우 초기화하기 위해 필요)
• 임시 좌표 nx, ny를 만들어 모든 조건을 만족했을 경우에 현재 좌표를 변경

x좌표, y좌표 나오면 일단 헷갈리기 시작....다시 차근차근 생각해보겠다며 코드 12938091번 수정했는데 개인적으로 첫 번째 풀이가 좀 더 이해하기 쉬웠다. 두 번째 풀이는 속도가 조금 더 빠르지만 머리가 따라주지 않아서 힘들었다...휴...어쨌든 풀었으니까 쉬어야겠다.공원 산책을 왜이렇게 빡세게 하시는 거에요... 🏃🏻 🏃🏻