力扣-剑指 Offer 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

示例2

输入：m = 3, n = 1, k = 0

输出：1

提示

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

解题思路
递推
  定义 $vis[i][j]$ 为 $(i,j)$ 坐标是否可达，如果可达返回 $1$，否则返回 $0$。首先 $(i,j)$ 本身需要可以进入，因此需要先判断 $i$和 $j$ 的数位之和是否大于 $k$ ，如果大于的话直接设置 $vis[i][j]$ 为不可达即可。

  否则，前面提到搜索方向只需朝下或朝右，因此 $(i,j)$ 的格子只会从 $(i-1,j)$ 或者 $(i,j-1)$ 两个格子走过来（不考虑边界条件），那么 $vis[i][j]$ 是否可达的状态则可由如下公式计算得到：
$$vis[i][j]=vis[i-1][j]orvis[i][j-1]$$
  即只要有一个格子可达，那么 $(i,j)$ 这个格子就是可达的，因此我们只要遍历所有格子，递推计算出它们是否可达然后用变量 $ans$ 记录可达的格子数量即可。

初始条件 $vis[i][j]=1$ ，递推计算的过程中注意边界的处理。

python3代码

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        def getDigitSum(i,j):
            res=0
            while i:
                res+=i%10
                i //=10
            while j:
                res +=j%10
                j //=10
            return res
        visited=set()
        def dfs(i,j):
            if i>=m or j>=n or getDigitSum(i,j)>k or (i,j)in visited:
                return 0
            visited.add((i,j))
            return 1+dfs(i,j+1)+dfs(i+1,j)
        return dfs(0,0)

java代码

class Solution {
    public int movingCount(int m, int n, int k) {
        if(k==0)
            return 1;
        boolean [][] vis=new boolean[m][n];
        int ans=1;
        vis[0][0]=true;
        for(int i=0;i<m;i++){
             for(int j=0;j<n;j++){
                 if((i==0&&j==0)|| get(i)+get(j)>k){
                     continue;
                 }
                 if(i-1>=0){
                     vis[i][j] |=vis[i-1][j];
                 }
                 if(j-1>=0){
                     vis[i][j] |=vis[i][j-1];
                 }
                 ans +=vis[i][j] ?1:0;

             }
        }
        return ans; 
           
    }
    private int get(int x){
        int res=0;
        while(x!=0){
            res +=x%10;
            x /=10;
        }
        return res;
    }
}