题目要求:地上有一个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;
	}
Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐