题目
有一个X*Y的网格,一个机器人只能走格点且只能向右或向下走,要从左上角走到右下角。请设计一个算法,计算机器人有多少种走法。注意这次的网格中有些障碍点是不能走的。
给定一个int[][] map(C++ 中为vector >),表示网格图,若map[i][j]为1则说明该点不是障碍点,否则则为障碍。另外给定int x,int y,表示网格的大小。请返回机器人从(0,0)走到(x - 1,y - 1)的走法数,为了防止溢出,请将结果Mod 1000000007。保证x和y均小于等于50
解答
方法1: 递归穷举(超时),利用递归计算出每次的结果,但是超时,不能AC,代码如下:
import java.util.*;
public class Robot {
public int countWays(int[][] map, int x, int y) {
// write code here
return count(map,x,y,0,0);
}
public int count(int[][] map, int x, int y,int i,int j){
int co = 0;
if(i == x - 1&&j == y - 1)
return 1;
if(i < x - 1&&map[i + 1][j] == 1)
co += count(map,x,y,i + 1,j);
if(j < y - 1&&map[i][j + 1] == 1)
co += count(map,x,y,i,j + 1);
return co;
}
}
方法2:利用DP去解,DP是通过求子问题的最优解去求出问题的最优解,在这个问题中就等于动规打表,dp数组里的值为到达这个格子有多少种解法,然后通过各种约束条件去循环求解,例如当遇到map数组为1时,表示这个点无法到达,则开始下一轮循环,以及起点和贴着边界走的情况。最终,循环过后,dp数组的最后一个元素则为本题的解。
import java.util.*;
public class Robot {
public int countWays(int[][] map, int x, int y) {
// write code here
int[][] dp = new int[x][y];
for(int i = 0;i < x;i++){
for(int j = 0;j < y;j++){
if(map[i][j] != 1){
continue;
}
if(i == 0&& j== 0){
dp[i][j] = 1;
}
else if(i == 0&&j != 0){
dp[i][j] = dp[i][j - 1];
}
else if(i != 0&&j == 0){
dp[i][j] = dp[i - 1][j];
}
else{
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
}
return dp[x - 1][y - 1]%1000000007;
}
}