机器人的运动范围

题目描述

地上有一个 m 行和 n 列的方格，横纵坐标范围分别是 0∼m−1 和 0∼n−1。

一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格。

但是不能进入行坐标和列坐标的数位之和大于 k的格子。

请问该机器人能够达到多少个格子？

样例1

输入：k=7, m=4, n=5

输出：20

样例2

输入：k=18, m=40, n=40

输出：1484

解释：当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。

　　　但是，它不能进入方格（35,38），因为3+5+3+8 = 19。

注意:

    1. 0<=m<=50
    2. 0<=n<=50
    3. 0<=k<=100

思路

　　dfs，遇到一个能进入的方格ans就加1，然后判断它周围四个方向的格子能不能进入、之前有没有访问过，从而判断是否要进入并继续dfs，若它四个方向都不可行，那么就return，最后输出ans即可.

※温馨提示

　　1. 当m=0或n=0时，ans=0；

　　2. 若没有可行路径能到达格子p，那即便p的位数和小于k也没用，是到大不了的（一开始没注意到这点，所以直接二层循环，就错了QAQ）；

class Solution {
public:
    int d[4][2]={{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
    bool vis[55][55];
    int ans=0;
    
    void dfs(int k, int m, int n, int x, int y) {
        int xx=x, yy=y, sum=0;
        while(xx) {
            sum+=xx%10;
            xx/=10;
        }
        while(yy) {
            sum+=yy%10;
            yy/=10;
        }
        if(sum<=k) {
            ans++;
            vis[x][y]=true;
            for(int i=0; i<4; i++) {
                int dx=x+d[i][0];
                int dy=y+d[i][1];
                if(dx>=0 && dx<m && dy>=0 && dy<n && !vis[dx][dy]) dfs(k, m, n, dx, dy);
            }
        }
        return;
    }
    
    int movingCount(int threshold, int rows, int cols)
    {
        if(rows>0 && cols>0)
            dfs(threshold, rows, cols, 0, 0);
        return ans;
    }
};