bug
别忘了对走过的位置进行标记,防止重走死循环
优化
- 仅走下和右两个方向
- 基于上个位置的数位和,计算当前位置的数位和
难度中等276
地上有一个m行n列的方格,从坐标 [0,0]
到坐标 [m-1,n-1]
。一个机器人从坐标 [0, 0]
的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1 输出:3
示例 2:
输入:m = 3, n = 1, k = 0 输出:1
1、最原始的 dfs(四个方向、每次重新计算数位和)
//思路:dfs,注意:对走过的位置进行记录,防止重复进入
int result;
public int movingCount(int m, int n, int k) {
result=0;
boolean[][] visited=new boolean[m][n];
dfs(0,0,m,n,k,visited);
return result;
}
public void dfs(int i,int j,int m, int n,int k,boolean[][] visited){
if(i>(m-1)||j>(n-1)||i<0||j<0||visited[i][j]||(i%10+i/10+j%10+j/10)>k) return ;
result++;
//bug!!!别忘了对走过的位置进行标记,防止重走死循环
visited[i][j]=true;
//int[][] ds={{-1,0},{1,0},{0,-1},{0,1}};
//优化1:
int[][] ds={{1,0},{0,1},{-1,0},{0,-1};
for(int[] d:ds){
int p=i+d[0];
int q=j+d[1];
dfs(p,q,m,n,k,visited);
}
}
2、跳出题目的坑(四个方向),仅走下和右两个方向,且基于上个位置的数位和,计算当前位置的数位和
int result,m,n,k;
public int movingCount(int m, int n, int k) {
this.result=0;
this.m=m;
this.n=n;
this.k=k;
boolean[][] visited=new boolean[m][n];
dfs(0,0,0,0,visited);
return result;
}
//优化2
public void dfs(int i,int j,int si, int sj,boolean[][] visited){
if(i>(m-1)||j>(n-1)||i<0||j<0||(si+sj)>k||visited[i][j]) return ;
result++;
visited[i][j]=true;
dfs(i+1,j,(i+1)%10==0?si-8:si+1,sj,visited);
dfs(i,j+1,si,(j+1)%10==0?sj-8:sj+1,visited);
}