机器人的运动范围
题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
解题思路
- 只要从(0,0)为起点遍历,对每个位置访问其四周,标记数组标记是否被访问过
- 对每个合法的位置,接着对他的四周进行访问
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
if(threshold<=0 || rows<=0||cols<=0)
return 0;
bool* visited=new bool[rows*cols];
memset(visited,false,rows*cols);
//只从(0,0)位置开始遍历
int ans=func(threshold,rows,cols,0,0,visited);
delete[] visited;
return ans;
}
int func(int threshold,int rows,int cols,int i,int j,bool* visited){
int cnt=0;
//没有出界&&当前位置未访问&&当前行列坐标数位之和不大于k
if(i>=0&&j>=0&&i<rows&&j<cols && visited[i*cols+j]==false&&digitsum(i)+digitsum(j)<=threshold){
visited[i*cols+j] = true;
cnt = 1+func(threshold,rows,cols,i+1,j,visited)
+func(threshold,rows,cols,i-1,j,visited)
+func(threshold,rows,cols,i,j+1,visited)
+func(threshold,rows,cols,i,j-1,visited);
}
return cnt;
}
//计算数位之和
int digitsum(int a){
int sum=0;
while(a>0){
sum=sum+a%10;
a/=10;
}
return sum;
}
};