剑指offer--机器人的运动范围

题目描述

地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

AC代码

典型的回溯法就可以解决

public class Solution {
    int count = 0;  //保存最终的结果，即可以到达多少个格子

    public int movingCount(int threshold, int rows, int cols)
    {
        boolean[][] seen = new boolean[rows][cols];  //判断是否被遍历过
        backtrack(threshold, rows, cols, seen, 0, 0); //从 [0][0] 开始遍历
        return count;
    }

    //direction
    int[] dirx = {0, 0, -1, 1};
    int[] diry = {1, -1, 0, 0};

    private void backtrack(int threshold, int rows, int cols, boolean[][] seen, int i, int j){
        if(i < 0 || i >= rows || j < 0 || j >= cols || seen[i][j] || (sumBit(i) + sumBit(j)) > threshold) return;
        seen[i][j] = true;
        count++;
        //从当前节点开始，依次遍历上下左右四个方向（方向的顺序对结果没有影响）
        for(int k = 0;k < 4;k++)
            backtrack(threshold, rows, cols, seen, i + dirx[k], j + diry[k]);
        //seen[i][j] = false;
    }

    private int sumBit(int i){
        int sum = 0;
        while(i != 0){
            sum += i % 10;
            i /= 10;
        }
        return sum;
    }
}