机器人的运动范围 -剑指offer中等

机器人的运动范围
题目链接
地上有一个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

解题思路

• 深度优先搜索
• 不合规坐标：越界坐标以及坐标之和大于k的不能走

解题代码

class Solution {
public:
    int dx[4] = {0,0,1,-1};
    int dy[4] = {-1,1,0,0};
    int ans=0;
    bool vis[101][101];
    bool judge(int x, int y, int k){
        int s=0;
        while(x){
            s+=x%10;
       
            x/=10;
        }
        while(y){
            s+=y%10;
            y/=10;
        }
        return s>k;
    }
    void dfs(int m, int n, int k, int x, int y){
        if(x==m-1&&y==n-1)return;
        for(int i=0;i<4;i++){
            int nx = dx[i]+x;
            int ny = dy[i]+y;
            if(nx<0||ny<0||nx>=m||ny>=n||vis[nx][ny]||judge(nx,ny,k))continue;
            ans+=1;
            vis[nx][ny]=1;
            dfs(m,n,k,nx,ny);
        }
    }
    int movingCount(int m, int n, int k) {
        if(k<0)return 0;
        vis[0][0]=1;
        ans+=1;
        dfs(m,n,k,0,0);
        return ans;
    }
};