地上有一个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。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:输入:m = 3, n = 1, k = 0
输出:1
提示:1 <= n,m <= 100
0 <= k <= 20
一道很基础的动态规划题目,做这种移动问题前我们应该想到这种题的做法其实都是一样的没什么难的,首先定义一个移动矩阵是我们最先想到的,这是毫无疑问的
int move[4][2]={{0,1},{0,-1},{-1,0},{1,0}}; //定义运动矩阵
由于这道题只是简单的上下左右,并没有涉及到左下左上,右下右上这另外四种移动方式,就很简单了,首先题目说到的该坐标的和不能大于k值,而这个坐标和并不是简单的相加而是进行的分裂开在和到一起,所以我们编一个函数来运算这样显得更美观一点,这样我们计算坐标和的时候我们就可以直接sum(x)+sum(y)与k进行比大小了
int sum(int x)
{
int s=0;
while(x>0)
{
s+=x%10;
x=x/10;
}
return s;
}
然后第二部就是看边界了,横坐标不能大于m不能小于0,纵坐标不能大于n不能小于0,我们也将其封装成为一个函数
bool flag(int x,int y,int m,int n)
{
if(x>=0&&x<m&&y>=0&&y<n)
{
return true;
}else{
return false;
}
}
最后一步就是动态规划的思想了,与上一步有关系我们只要找到上一步是否可以走,然后根据上面的两个条件就可以判断这条路能不能走了:
if(sum(p)+sum(q)<=k&&flag(p,q,m,n)&&arr[i][j]==1)
{
arr[p][q]=1;
}
按照上面的步骤我们就可以得到一个只带有0.1得矩阵我们设置一个值把1得全部加起来就得到最终得结果了
最后我们看完整代码:
int sum(int x)
{
int s=0;
while(x>0)
{
s+=x%10;
x=x/10;
}
return s;
}
bool flag(int x,int y,int m,int n)
{
if(x>=0&&x<m&&y>=0&&y<n)
{
return true;
}else{
return false;
}
}
int movingCount(int m, int n, int k){
int move[4][2]={{0,1},{0,-1},{-1,0},{1,0}}; //定义运动矩阵
int i,j,arr[m][n],x,y,max=0,p=0,q=0;
memset(arr,0,sizeof(arr));
arr[0][0]=1;
for(i=0;i<m;i++)
{
for(j=0;j<n;j++)
{
for(x=0;x<4;x++)
{
for(y=0;y<2;y++)
{
p=i+move[x][0],q=j+move[0][y];
if(sum(p)+sum(q)<=k&&flag(p,q,m,n)&&arr[i][j]==1)
{
arr[p][q]=1;
}
}
}
}
}
for(i=0;i<m;i++)
{
for(j=0;j<n;j++)
{
if(arr[i][j]==1)
{
max++;
}
}
}
return max;
}
看最后运行结果图(当然我也是一次通过得,因为很久没写算法了,差不多都快半年了,现在每天晚上坚持写一道leetcode题吧,并且发布博客)