题目要求:地上有一个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。请问该机器人能够到达多少个格子?
显然要遍历所有的网格直到边界条件为止。这里使用了深度优先搜索来遍历网络。此外需要注意的是遍历过程中的重复位置。可以使用二维数组的标志位来记录搜索途径,也可以记录每一个搜索过的坐标,每新搜索到一个坐标时来看看是否搜索过。
void dfs(int k,int m,int n,vector<pair<int,int> > &res,int i,int j)
{
if((i>=0&&i<m)&&(j>=0&&j<n)&&((i%10+i/10+j%10+j/10)<=k))
{
int cnt=0;
for(auto a:res)
{
if(a.first ==i&&a.second ==j)
return;
cnt++;
}
if(cnt==res.size ())
res.push_back (make_pair(i,j));
dfs(k,m,n,res,i,j+1);
dfs(k,m,n,res,i+1,j);
}
else
return ;
}
int movingCount(int threshold, int rows, int cols)
{
vector<pair<int,int> > res;
dfs(threshold,rows,cols,res,0,0);
int sum=res.size();
return sum;
}