地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?


递推解法

class Solution {
    private int res = 0;

    public int movingCount(int m, int n, int k) {
        int [][] visit = new int[m][n];
        visit[0][0] = 1;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (check(i,j,k)){
                    if (i>0){//如果上面的格子为1,则说明这个可以从上面这个格子下来
                        visit[i][j] |= visit[i-1][j];
                    }
                    if (j>0){//使用或等于是防止在这个地方出现更改
                        visit[i][j] |= visit[i][j-1];
                    }
                    res+=visit[i][j];
                }
            }
        }
        return res;
    }




    private boolean check(int i, int j, int k) {
        int count = 0;
        int t = i;
        while (t>0){
            count+=t%10;
            t = t/10;
        }
        t = j;
        while (t>0){
            count+=t%10;
            t = t/10;
        }
        return count <= k;
    }
}

dfs解法

class Solution {
    private int res = 0;

    private int m;
    private int n;
    private int k;

    public int movingCount(int m, int n, int k) {
        int [][] visit = new int[m][n];
        this.m=m;
        this.n = n;
        this.k = k;
        dfs(0,0,visit);
        return res;
    }

    private void dfs(int i, int j, int[][] visit) {
        if (i>=m || j>=n || visit[i][j]==1){//判断边界条件
            return;
        }
        if (check(i,j,k)){
            res++;
            visit[i][j]=1;
            dfs(i+1,j,visit);//不需要回溯
            dfs(i,j+1,visit);
        }

    }


    private boolean check(int i, int j, int k) {
        int count = 0;
        int t = i;
        while (t>0){
            count+=t%10;
            t = t/10;
        }
        t = j;
        while (t>0){
            count+=t%10;
            t = t/10;
        }
        return count <= k;
    }
}