[python] 프로그래머스 - 공 이동 시뮬레이션(월간 코드 챌린지 시즌3)

🤔문제 해결

쿼리는 거꾸로 시작

끝이면 늘리고

끝이 아니면 이동

시작 시 범위

명령: 위로 한칸 (아래에 있어야 위로 왔을 때 별에 도착할 수 있다 or 제자리여도 위로 한칸이면 어짜피 끝이기 때문에 별에 도착할 수 있다)

아래로 한칸 늘려줌

명령: 왼쪽으로 한칸 (오른쪽에 있어야 왼쪽으로 왔을 때 별에 도착할 수 있다.)

오른쪽으로 이동

명령: 위로 세칸 (아래쪽에 있어야 위로... 그러나 끝이기 때문에 늘릴 수 없다.)

그대로

명령: 오른쪽으로 한칸(왼쪽에 있어야 오른쪽으로 왔을 때 별에 도착가능

왼쪽으로 이동

명령: 위로 두칸 ( 아까 위로세칸인 경우랑 같음 ) 

그대로

명령: 아래로 한칸

그대로

answer = (x_max - x_min + 1) * (y_max - y_min + 1)

 

 

💨 중간에 범위를 아예 벗어나서 결과가 0인 경우가 발생한다.

 

💻소스 코드

def solution(n, m, x, y, queries):
    answer = 0
    x_min, x_max, y_min, y_max = x, x, y, y

    for idx in range(len(queries) - 1, -1, -1):
        direc, dist = queries[idx]
        if direc == 0:  # 좌(오른쪽에서 왔음)
            y_max += dist  # 오른쪽으로 늘리기
            if y_max > m - 1: # 범위 벗어나면
                y_max = m - 1  # 끝값으로
            if y_min != 0:  # 왼쪽 값이 끝이 아니면 범위 축소
                y_min += dist

        elif direc == 1:  # 우(왼쪽에서 왔음)
            y_min -= dist
            if y_min < 0:
                y_min = 0
            if y_max != m - 1:
                y_max -= dist

        elif direc == 2:  # 상(아래서 왔음)
            x_max += dist
            if x_max > n - 1:
                x_max = n - 1
            if x_min != 0:
                x_min += dist

        else:  # 하(위에서 왔음)
            x_min -= dist
            if x_min < 0:
                x_min = 0
            if x_max != n - 1:
                x_max -= dist
        if y_min > m - 1 or y_max < 0 or x_min > n - 1 or x_max < 0:
            return answer
    else:
         answer = (y_max - y_min + 1) * (x_max - x_min + 1)
    return answer

 

📕문제 확인

출처: 프로그래머스

링크: 공 이동 시뮬레이션