回溯法是指把所有的可能选项都放在一个系统当中,不断的尝试探索,直到找到最佳结果为止,回溯法最常用的就是递归的方法解决问题。下面我们通过机器人运动范围问题,详细解释回溯法。
问题:
地上有一个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
思路:
首先我们要先判断某一个值为0的情况,如果m,n,k中有一个值为0则返回0.然后我们要定义一个布尔类型的数组。用来记录每一步的执行状态。我们需要定义两个函数,第一个函数用来移动位置,更新布尔数组中对应的值,这个函数递归的遍历每个位置的上下左右邻居。第二个函数计算作标的位数和,这个结果在第一个函数中用来判断用。
代码:
class Solution {
public int movingCount(int m, int n, int k) {
if(m<=0||n<=0||k<0)
{
return 0;
}
boolean[] visited = new boolean[m * n];
for(int i=0;i<n*m;i++)
{
visited[i] = false;
}
int count = movingCountCore(m,n,0,0,k,visited);
return count;
}
public int movingCountCore(int m,int n,int row,int col,int k,boolean[] visited)
{
int count = 0;
if(0<=row&&row<m&&0<=col&&col<n&&(Getnumber(row)+Getnumber(col)<=k)&&(!visited[row*n+col]))
{
visited[row*n+col] = true;
count = 1 + movingCountCore(m,n,row-1,col,k,visited)+ movingCountCore(m,n,row,col-1,k,visited)+ movingCountCore(m,n,row+1,col,k,visited)+ movingCountCore(m,n,row,col+1,k,visited);
}
return count;
}
public int Getnumber(int num)
{
int s = 0;
while(num>0)
{
s +=num%10;
num = num/10;
}
return s;
}
}