[프로그래머스 / DP] 등굣길 (java)

문제 설명

계속되는 폭우로 일부 지역이 물에 잠겼습니다. 물에 잠기지 않은 지역을 통해 학교를 가려고 합니다. 집에서 학교까지 가는 길은 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개 이하입니다.
• 집과 학교가 물에 잠긴 경우는 입력으로 주어지지 않습니다.

입출력 예

mnpuddlesreturn

4

3

[[2, 2]]

4

입출력 예 설명

 

 

[문제접근]

 

[java 코드]

1차시도 - 실패 (테스트 케이스 절반 통과)

class Solution {
    public int solution(int m, int n, int[][] puddles) {
        int answer = 0;
        int[][] arr = new int[n+1][m+1];
        
        arr[1][1] = 1;
        
        for(int[] puddle : puddles){
            arr[puddle[1]][puddle[0]] = -1;
        }
        
        for(int i=2; i<n+1; i++){
        
                arr[i][1] = arr[i-1][1];
        }
        for(int i=2; i<m+1; i++){
           
                arr[1][i] = arr[1][i-1];
        }
        
        
        for(int i = 2; i < n+1; i++){
            for(int j = 2; j < m+1; j++){
                if(arr[i][j] < 0)
                    arr[i][j] = 0;
                else{
                    arr[i][j] = arr[i][j-1] + arr[i-1][j];
                }
                
            }
        }
        answer = arr[n][m];
        
        
        return answer;
    }
}

문제점: 이러면 1열/ 1행에 있는 웅덩이들이 무시된다 

 

[java 코드]

import java.util.Arrays;
class Solution {
    public int solution(int m, int n, int[][] puddles) {
        int answer = 0;
        int mod = 1000000007;
        int[][] arr = new int[n][m];

        for(int i = 0; i < puddles.length; i++) {
            arr[puddles[i][1] - 1][puddles[i][0] - 1] = -1; 
        }
        for(int i = 0; i < m; i++) {
            if(arr[0][i] == -1) {
                break;
            }
            arr[0][i] = 1;
        }
        for(int i = 0; i < n; i++) {
            if(arr[i][0] == -1) {
                break;
            }
            arr[i][0] = 1;
        }
        for(int i = 1; i < n; i++) {
            for(int j = 1; j < m; j++) {
                if(arr[i][j] == -1) {
                    continue;
                }
                if(arr[i - 1][j] != -1 && arr[i][j - 1] != -1) {                    
                    arr[i][j] = (arr[i - 1][j] + arr[i][j - 1]) % mod;
                }else if(arr[i - 1][j] != -1) {
                    arr[i][j] = arr[i - 1][j];
                }else if(arr[i][j - 1] != -1) {
                    arr[i][j] = arr[i][j - 1];
                }
            }
        }
        answer = arr[n - 1][m - 1];
        return answer;
    }
}

 

[또 다른 코드]

class Solution {
    public int solution(int m, int n, int[][] puddles) {
        int answer = 0;

        int[][] dp = new int[n + 1][m + 1];

        dp[1][1] = 1;
        for (int[] puddle : puddles)
            dp[puddle[1]][puddle[0]] = -1;

        for (int i = 1; i < dp.length; i++) {
            for (int j = 1; j < dp[i].length; j++) {
                if (dp[i][j] == -1) {
                    dp[i][j] = 0;
                } else {
                    if (i == 1) {
                        dp[i][j] += dp[i][j - 1];
                        // 0 라인은 0으로 초기화되어있기 때문에.. 
                    } else {
                        dp[i][j] = (dp[i - 1][j] + dp[i][j - 1]) % 1000000007;
                    }
                }
            }
        }
        answer = dp[n][m];

        return answer;
    }
}

[참고]

https://jar100.tistory.com/27

프로그래머스 알고리즘 : 등굣길 (java)

[문제 링크]

https://programmers.co.kr/learn/courses/30/parts/12263

프로그래머스