面试题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。请问该机器人能够到达多少个格子？

 

示例 1：

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

输入：m = 3, n = 1, k = 0
输出：1
提示：

1 <= n,m <= 100
0 <= k <= 20

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路：

1.从(0,0)开始移动，判断机器人能够进入方格 (0,0)

2.向左、右、上、下移动一格时，是否满足行坐标和列坐标的数位之和<=k，若满足，则表示机器人可以走过这些格子

3.将所有可以到达的格子相加，返回即可

代码实现：

class Solution {
public:
    int movingCount(int m, int n, int k) {
        bool *visited=new bool[m*n];
        memset(visited,0,m*n);
        int cnt=countSpace(m,n,k,0,0,visited);
        delete[] visited;
        return cnt;
    }

    int countSpace(int m,int n,int k,int i,int j,bool*visited){//该机器人能够到达多少个格子
        int count=0;
        if(check(m,n,k,i,j,visited)){//检查当前(i,j)是否可以进入
            visited[i*n+j]=true;
            count=1+countSpace(m,n,k,i-1,j,visited)
            +countSpace(m,n,k,i,j-1,visited)
            +countSpace(m,n,k,i+1,j,visited)
            +countSpace(m,n,k,i,j+1,visited);
        }
        return count;
    }

    bool check(int m,int n,int k,int i,int j,bool*visited){
        if(i>=0&&i<m&&j>=0&&j<n&&!visited[i*n+j]&&(digit(i)+digit(j)<=k)){
            return true;
        }
        return false;
    }

    int digit(int num){
        int sum=0;
        while(num){
            sum+=num%10;
            num/=10;
        }
        return sum;
    }
};

 

耗时和内存使用情况：