LeetCode面试题13：机器人的运行范围

题目描述

地上有一个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。请问该机器人能够到达多少个格子？

示例

输入：m = 2, n = 3, k = 1
输出：3

答案

解法一：

深度优先遍历（dfs）的递归方法，代码如下：
写法一：

class Solution {
    public int movingCount(int m, int n, int k) {
        boolean[][] visited = new boolean[m][n];
        dfs(0,0,m,n,k,visited);
        int result = 0;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(visited[i][j]){
                    result++;
                }
            }
        }
        return result;
    }
    public void dfs(int i,int j,int m,int n,int k,boolean[][] visited){
        if(i >= m || i<0 || j >= n || j < 0 || visited[i][j]){
            return;
        }
        if(canEnter(i,j,k)){
            visited[i][j] = true;
            //左上角[0][0]位置为起始位置，对其它任何一个位置，它要么从上面一个位置到达，要么从左面一个位置到达。
            dfs(i+1,j,m,n,k,visited);
            //dfs(i-1,j,m,n,k,visited);
            dfs(i,j+1,m,n,k,visited);
            //dfs(i,j-1,m,n,k,visited);
        }else{
            return;
        }

    }
    public boolean canEnter(int i,int j,int k){
        int sum = 0;
        while(i > 9){
            sum = sum + i%10;
            i = i/10;
        }
        sum = sum + i;
        while(j > 9){
            sum = sum + j%10;
            j = j/10;
        }
        sum = sum + j;
        return sum <= k;
    }
}

写法二：

class Solution {
    public int movingCount(int m, int n, int k) {
        boolean[][] visited = new boolean[m][n];
        return recursive(m,n,0,0,k,visited);
    }
    public int recursive(int m,int n,int i,int j,int k,boolean[][] visited){
        if(i<0||i>=m||j<0||j>=n||!canEnter(i,j,k)||visited[i][j]){
            return 0;
        }
        visited[i][j] = true;
        return recursive(m,n,i+1,j,k,visited)+recursive(m,n,i,j+1,k,visited)+1;

    }
    public boolean canEnter(int i,int j,int k){
        int res = 0;
        while(i/10!=0){
            res = res + i/10;
            i = i%10;
        }
        res = res + i;
        while(j/10!=0){
            res = res + j/10;
            j = j%10;
        }
        res = res + j;
        return res <= k;
    }
}

解法二：

递推的方法
左上角[0][0]位置为起始位置，对其它任何一个位置，它要么从上面一个位置到达，要么从左面一个位置到达。使用visited记录是否可达，如果某个位置的左或者上，都不可达，那么该位置不可达。如果左或者上有至少一个可达，那么判断该位置的行列坐标是否符合要求即可。代码如下：

class Solution {
    public int movingCount(int m, int n, int k) {
        if(m<=0||n<=0||k<0){
            return 0;
        }
        boolean[][] visited = new boolean[m][n];
        int result = 0;
        //初始化左上角第一个位置
        visited[0][0] = canEnter(0,0,k);
        result = visited[0][0]?result+1:result;
        //第一行
        for(int j=1;j<n;j++){
            visited[0][j] = visited[0][j-1]&&canEnter(0,j,k);
            result = visited[0][j]?result+1:result;
        }
        //第一列
        for(int i=1;i<m;i++){
            visited[i][0] = visited[i-1][0]&&canEnter(i,0,k);
            result = visited[i][0]?result+1:result;
        }
        //其它
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                visited[i][j] = (visited[i-1][j]||visited[i][j-1])&&canEnter(i,j,k);
                result = visited[i][j]?result+1:result;
            }
        }
        return result;
    }
    
    public boolean canEnter(int i,int j,int k){
        int res = 0;
        while(i/10!=0){
            res = res + i/10;
            i = i%10;
        }
        res = res + i;
        while(j/10!=0){
            res = res + j/10;
            j = j%10;
        }
        res = res + j;
        return res <= k;
    }
}