剑指 Offer 13. 机器人的运动范围--慢慢改进解(中等)Java

题目描述

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

示例：

思路：

思路一：

创建一个新的数组，通过递归的方式从0，0 开始向下、向右递归，符合条件该数组值为1，最后统计数组和。但是时间超时了。

思路二：

原因分析，是以为递归中有重复递归过的位置，所以计算时间长。通过初始化判断该位置是否为初始化值，来进行递归，原则上复杂度降为O(m*n)。

思路三：

看了大神们的代码，受到了启发，将数组改为 boolean 型，这样不用再循环遍历，时间服大肚上大致为思路二的1/2。

代码：

思路一代码：

class Solution {
    public int movingCount(int m, int n, int k) {
        if(k == 0){
            return 1;
        }
        int[][] arr = new int[m][n];
        int sum = 0;//定义初始能到达格子数
        int a = 0;
        int b = 0;
        recursion(m,n,k,a,b,arr);
        for(int i = 0;i<m;i++){
            for(int j = 0;j<n;j++){
                sum += arr[i][j];
            }
        }
        return sum;
    }
    //向下，向右递归
    public void recursion(int m,int n,int k,int a,int b,int[][] arr){
        if(isTrue(a,b,k)){
            arr[a][b] = 1;
            if(a < m-1){
                recursion(m,n,k,a+1,b,arr);
            }
            if(b<n-1){
                recursion(m,n,k,a,b+1,arr);
            }
        }
    }
    public boolean isTrue(int m,int n,int k){
        int sum = 0;
        while(m>0){
            sum += m % 10;
            m = m / 10;
        }
        while(n>0){
            sum += n % 10;
            n = n / 10;
        }
        if(sum<=k){
            return true;
        }else{
            return false;
        }
    }
}

思路二代码：

class Solution {
    public int movingCount(int m, int n, int k) {
        if(k == 0){
            return 1;
        }
        int[][] arr = new int[m][n];
        int sum = 0;//定义初始能到达格子数
        int a = 0;
        int b = 0;
        recursion(m,n,k,a,b,arr);
        for(int i = 0;i<m;i++){
            for(int j = 0;j<n;j++){
                sum += arr[i][j];
            }
        }
        return sum;
    }
    //向下，向右递归
    public void recursion(int m,int n,int k,int a,int b,int[][] arr){
        if(isTrue(a,b,k) && arr[a][b]==0){
            arr[a][b] = 1;
            if(a < m-1){
                recursion(m,n,k,a+1,b,arr);
            }
            if(b<n-1){
                recursion(m,n,k,a,b+1,arr);
            }
        }
    }
    public boolean isTrue(int m,int n,int k){
        int sum = 0;
        while(m>0){
            sum += m % 10;
            m = m / 10;
        }
        while(n>0){
            sum += n % 10;
            n = n / 10;
        }
        if(sum<=k){
            return true;
        }else{
            return false;
        }
    }
}

思路三代码：

class Solution {
    int sum = 0;//定义初始能到达格子数
    public int movingCount(int m, int n, int k) {
        if(k == 0){
            return 1;
        }
        boolean[][] arr = new boolean[m][n];
        int a = 0;
        int b = 0;
        recursion(m,n,k,a,b,arr);
        return sum;
    }
    //向下，向右递归
    public void recursion(int m,int n,int k,int a,int b,boolean[][] arr){
        if(isTrue(a,b,k) && arr[a][b]==false){
            arr[a][b] = true;
            sum++;
            if(a < m-1){
                recursion(m,n,k,a+1,b,arr);
            }
            if(b<n-1){
                recursion(m,n,k,a,b+1,arr);
            }
        }
    }
    public boolean isTrue(int m,int n,int k){
        int sum = 0;
        while(m>0){
            sum += m % 10;
            m = m / 10;
        }
        while(n>0){
            sum += n % 10;
            n = n / 10;
        }
        if(sum<=k){
            return true;
        }else{
            return false;
        }
    }
}

结果：

思路二结果：

思路三结果：