原文地址:https://www.jianshu.com/p/4df50a647831
时间限制:1秒 空间限制:32768K
题目描述
地上有一个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)
{
if(threshold<0 || rows<1 ||cols<1)
return 0;
int* flag=new int[rows*cols]();
return movingCountHelper(threshold,rows,cols,flag,0,0);
}
int movingCountHelper(int threshold, int rows, int cols, int* flag, int i,int j){
if(i<0 || i>=rows || j<0 || j>=cols
|| flag[i*cols+j]==1 || mySum(i)+mySum(j)>threshold)
return 0;
flag[i*cols+j]=1;
return 1+movingCountHelper(threshold,rows,cols,flag,i+1,j)
+movingCountHelper(threshold,rows,cols,flag,i-1,j)
+movingCountHelper(threshold,rows,cols,flag,i,j+1)
+movingCountHelper(threshold,rows,cols,flag,i,j-1);
}
int mySum(int x){
int res=0;
while(x){
res+=x%10;
x/=10;
}
return res;
}
};
运行时间:5ms
占用内存:504k