프로그래머스 등굣길 파이썬 풀이 - 다이나믹프로그래밍(DP)

문제 출처

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

프로그래머스

문제

문제 설명

계속되는 폭우로 일부 지역이 물에 잠겼습니다. 물에 잠기지 않은 지역을 통해 학교를 가려고 합니다. 집에서 학교까지 가는 길은 m x n 크기의 격자모양으로 나타낼 수 있습니다.

아래 그림은 m = 4, n = 3 인 경우입니다.

가장 왼쪽 위, 즉 집이 있는 곳의 좌표는 (1, 1)로 나타내고 가장 오른쪽 아래, 즉 학교가 있는 곳의 좌표는 (m, n)으로 나타냅니다.

격자의 크기 m, n과 물이 잠긴 지역의 좌표를 담은 2차원 배열 puddles이 매개변수로 주어집니다. 오른쪽과 아래쪽으로만 움직여 집에서 학교까지 갈 수 있는 최단경로의 개수를 1,000,000,007로 나눈 나머지를 return 하도록 solution 함수를 작성해주세요.

 

제한사항

• 격자의 크기 m, n은 1 이상 100 이하인 자연수입니다.
  • m과 n이 모두 1인 경우는 입력으로 주어지지 않습니다.
• 물에 잠긴 지역은 0개 이상 10개 이하입니다.
• 집과 학교가 물에 잠긴 경우는 입력으로 주어지지 않습니다.

입출력 예

m	n	puddles	return
4	3	[[2, 2]]	4

입출력 예 설명

문제 풀이

지도 = 2차원 좌표 평면

puddles 위치가 뒤집어져 있으니까 좌우의 값을 바꾼다

지도를 하나하나 지나가면서 지나온 값들을 더한다.

만약 지나가다가 웅덩이가 있으면 값을 0으로 초기화 한다.(지나가면 안되니까)

최소 길의 갯수를 1,000,000,007로 나눈 나머지를 return 한다.

 

소스코드

DIV_NUM = 1000000007
def solution(m, n, puddles):
    dp = [[0]*(m+1) for _ in range(n+1)]
    dp[1][1] = 1
    for i, j in puddles:
        dp[j][i] = -1
    
    for i in range(1, n+1):
        for j in range(1, m+1):
            if dp[i][j] == -1:
                dp[i][j] = 0
                continue
            dp[i][j] += (dp[i-1][j] + dp[i][j-1])
    if dp[n][m] >= DIV_NUM:
        dp[n][m] %= DIV_NUM
    return dp[n][m]