题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:
这种问题用回溯法
认真分析最终条件,最终条件一旦找到,就很好做了
本题的最终条件就是,当遍历了所有可以进入的格子后就返回。
代码:
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
bool *visited=new bool[rows*cols];
memset(visited,false,rows*cols);
int count=moving(rows,cols,0,0,threshold,visited);
delete[]visited;
return count;
}
int moving(int rows,int cols,int row,int col,int k,bool *visited)
{
int count=0;
if(row>=0&&row<rows&&col>=0&&col<cols&&smallthanK(row,col,k)&&!visited[row*cols+col])
{
visited[row*cols+col]=true;
count=1+moving(rows,cols,row-1,col,k,visited)+
moving(rows,cols,row+1,col,k,visited)+
moving(rows,cols,row,col-1,k,visited)+
moving(rows,cols,row,col+1,k,visited);
}
return count;
}
bool smallthanK(int row,int col,int k)
{
int sum=0;
while(row>0)
{
sum+=row%10;
row=row/10;
}
while(col>0)
{
sum+=col%10;
col=col/10;
}
if(sum<=k)
return true;
return false;
}
};
总结:通常情况下,物体或人在二维方格运动的这类问题都可以用回溯法解决