13-机器人的运动范围
大约 2 分钟
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
题目中的可到达的格子是可以移动到的格子,假如k=1,那么(10,0)虽然满足不大于k,但是由于(9,0),(11,0),(10,1)都无法到达,所以(10,0)也是无法到达的。
那么我们参照12-矩阵中的路径的解法,依然是走迷宫的思路,我们找一个点,开始走,如果下一个格子可以通过,那么继续找这个格子的四周的格子,然后循此往复,直到再也没有可以走的格子。这实际就是深度优先搜索的方法。
以下面的代码为例
public int movingCount(int k, int rows, int cols)
{
if(k == 0){
return 1;
}
boolean[] visited = new boolean[rows*cols];
int count = movingCount(rows,cols,k,0,0,visited);
return count;
}
public int movingCount(int rows,int cols,int k,int row, int col,boolean[] visited){
int count = 0;
int index = row*cols+col;
// 如果格子在矩阵范围内 同时没有访问过 同时符合要求
if(row >= 0 && col >= 0 && row < rows && col < cols && !visited[index] && (getDigitSum(row)+getDigitSum(col)) <= k){
//进行选择 标记已经访问过了
visited[index] = true;
//继续访问结点周围的结点
count = 1 + movingCount(rows,cols,k,row-1,col,visited)
+movingCount(rows,cols,k,row+1,col,visited)
+movingCount(rows,cols,k,row,col-1,visited)
+movingCount(rows,cols,k,row,col+1,visited);
}
return count;
}
public int getDigitSum(int num){
int sum = 0;
int rem = 0;
while(num != 0 ){
rem = num % 10;
sum += rem;
num /= 10;
}
return sum;
}