机器人的运动范围

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

 

java:

public class Solution {
    private int rows ;
    private int cols ;
    private int cnt = 0 ;
    private int threshold ;
    private static final int[][] next = {{0,1} , {0,-1} , {1,0} , {-1,0}} ;
    private int[][] digitSum ;
    public int movingCount(int threshold, int rows, int cols)
    {
        this.threshold = threshold ;
        this.rows = rows ;
        this.cols = cols ;
        init() ;
        boolean[][] mark = new boolean[rows][cols] ;
        dfs(0,0,mark) ;
        return cnt ;
    }
    
    private void dfs(int r, int c , boolean[][] mark){
        if (r < 0 || r >= rows || c < 0 || c >=cols || mark[r][c])
            return ;
        mark[r][c] = true ;
        if (digitSum[r][c] > threshold)
            return ;
        cnt++ ;
        for(int[] n : next){
            dfs(r+n[0],c+n[1],mark) ;
        }
    }
    
    private void init(){
        int[] digitOne = new int[Math.max(rows,cols)] ;
        for(int i = 0 ; i < digitOne.length ; i++){
            int n = i ;
            while(n > 0){
                digitOne[i] += n%10 ;
                n/=10 ;
            }
        }
        digitSum = new int[rows][cols] ;
        for(int i = 0 ; i < rows ; i++){
            for(int j = 0 ; j < cols ; j++){
                digitSum[i][j] = digitOne[i] + digitOne[j] ;
            }
        }
    }
}