题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?

【DFS】经典dfs: 先把不能走的格子标记为true

class Solution {
public:
    int n,m,k;
    bool f[1000][1000];
    int ans=0;
    int dir[4][2]={{0,1},{0,-1},{-1,0},{1,0}};
    int compute(int x,int y)
    {
        int sum=0;
        while(x)
        {
            sum+=x%10;
            x/=10;
        }
        while(y)
        {
            sum+=y%10;
            y/=10;
        }
        return sum;
    }
    void dfs(int x,int y)
    {
        if(x<0||x>n-1||y<0||y>m-1||f[x][y]) return;
        f[x][y]=true;
        ans+=1;
        for(int i=0;i<4;i++)
        {
            int dx=x+dir[i][0];
            int dy=y+dir[i][1];
            dfs(dx,dy);
        }
    }
    int movingCount(int threshold, int rows, int cols)
    {
        n=rows;
        m=cols;
        k=threshold;
        memset(f,false,sizeof(f));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(compute(i,j)>threshold) f[i][j]=true;
            }
        }
        dfs(0,0);
        return ans;
    }
};