【题目】:
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
【思路】:
采用回溯法,先判断机器人能否进入(i,j),再判断周围4个格子。
java code:
package com.offer.offer13;
/**
* @author zth
* @Date 2019-08-08 20:41
*/
public class RobotMove {
/**
* @param threshold 阈值
* @param rows 矩阵的行数
* @param cols 矩阵的列数
* @return 可达格子数
*/
public int movingCount(int threshold, int rows, int cols){
if (rows < 0 || cols < 0 || threshold<0){
return 0;
}
boolean[] isVisited = new boolean[rows*cols];
int count = movingCountCore(threshold,rows,cols,0,0,isVisited);
return count;
}
public int movingCountCore(int threshold, int rows, int cols,
int row,int col,boolean[] isVisited){
if (row<0 || col<0 || row>=rows || col>=cols || isVisited[row*cols+col]
|| getDigitSum(col)+getDigitSum(row) > threshold ){
return 0;
}
isVisited[row*cols+col]=true;
return 1+movingCountCore(threshold,rows,cols,row-1,col,isVisited)
+movingCountCore(threshold,rows,cols,row+1,col,isVisited)
+movingCountCore(threshold,rows,cols,row,col-1,isVisited)
+movingCountCore(threshold,rows,cols,row,col+1,isVisited);
}
private int getDigitSum(int num){
int res = 0;
while (num>0){
res += num%10;
num /=10;
}
return res;
}
// ========测试代码=========
void test(String testName, int threshold, int rows, int cols, int expected) {
if (testName != null)
System.out.print(testName + ":");
if (movingCount(threshold, rows, cols) == expected)
System.out.println("Passed.");
else
System.out.println("Failed.");
}
// 方格多行多列
void test1() {
test("Test1", 5, 10, 10, 21);
}
// 方格多行多列
void test2() {
test("Test2", 15, 20, 20, 359);
}
// 方格只有一行,机器人只能到达部分方格
void test3() {
test("Test3", 10, 1, 100, 29);
}
// 方格只有一行,机器人能到达所有方格
void test4() {
test("Test4", 10, 1, 10, 10);
}
// 方格只有一列,机器人只能到达部分方格
void test5() {
test("Test5", 15, 100, 1, 79);
}
// 方格只有一列,机器人能到达所有方格
void test6() {
test("Test6", 15, 10, 1, 10);
}
// 方格只有一行一列
void test7() {
test("Test7", 15, 1, 1, 1);
}
// 方格只有一行一列
void test8() {
test("Test8", 0, 1, 1, 1);
}
// 机器人不能进入任意一个方格
void test9() {
test("Test9", -10, 10, 10, 0);
}
public static void main(String[] args) {
RobotMove demo = new RobotMove();
demo.test1();
demo.test2();
demo.test3();
demo.test4();
demo.test5();
demo.test6();
demo.test7();
demo.test8();
demo.test9();
}
}