プログマーズ通学路

質問する

コード#コード#

cpp

#include <string>
#include <vector>

#define N 1000000007
using namespace std;

int solution(int m, int n, vector<vector<int>> puddles) {
    int answer = 0;
    vector<vector<int>> dp(n,vector<int>(m,0));
        
    for(auto i: puddles)
        dp[i[1]-1][i[0]-1]=-1;
    
    dp[0][0]=1;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if((i==0&&j==0)||dp[i][j]==-1) continue;
            
            if(dp[i-1][j]==-1||i==0) dp[i][j]=dp[i][j-1];
            else if(dp[i][j-1]==-1||j==0) dp[i][j]=dp[i-1][j];
            else dp[i][j]=dp[i-1][j]%N+dp[i][j-1]%N;  
        }
    }
    answer=dp[n-1][m-1]%N;
    return answer;
}

python

def solution(m, n, puddles):
    answer = 0
    dp = [[0]*(m+1) for _ in range(n+1)]
    dp[1][1] = 1
    for x, y in puddles:
        dp[y][x] = -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
            if i == j == 1:
                continue
            dp[i][j] = dp[i-1][j]+dp[i][j-1]

    return dp[n][m]%1000000007

に答える