剑指Offer——题13（机器人的运动范围)

【题目】：

地上有一个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();
    }

}