算法学习之机器人的运动范围

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

这个题目与上一道算法题类似。

使用dfs回溯法。

public class Solution {

    public int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<1||rows<1||cols<1){
            return 0;
        }
        boolean[][] visit = new boolean[rows][cols];
        return getCount(threshold,rows,cols,0,0,visit);
    }
    
    public int getCount(int threshold,int rows,int cols,int r,int c,boolean[][] visit){
        int count = 0;
        // r>=0&&c>=0&&r<rows&&c<cols行、列、在范围内大于0并在给定的范围
        // visit[r][c]不走重复格子
        // sum(r)+sum(c)<=threshold)坐标点和小于给定的阈值
        if(r>=0&&c>=0&&r<rows&&c<cols&&!visit[r][c]&&sum(r)+sum(c)<=threshold){
            visit[r][c] = true;
            // 因为从（0，0）开始，所以+1
            count = 1+
                getCount(threshold,rows,cols,r+1,c,visit)+
                getCount(threshold,rows,cols,r-1,c,visit)+
                getCount(threshold,rows,cols,r,c+1,visit)+
                getCount(threshold,rows,cols,r,c-1,visit);
        }
        return count;
    }
    public int sum(int number){
        int sum = 0;
        while(number>0){
            sum+=number%10; // 从个位开始获取每一位的总和如23%10为3，除完10在10%为2，2+3=5
            number/=10;
        }
        return sum;
    } 
}

时间复杂度：O(mn)

空间复杂度：O(mn)

练习地址：https://www.nowcoder.com/practice/6e5207314b5241fb83f2329e89fdecc8