題目描述

地上有一個m行和n列的方格。一個機器人從座標0,0的格子開始移動，每一次只能向左，右，上，下四個方向移動一格，但是不能進入行座標和列座標的數位之和大於k的格子。 例如，當k為18時，機器人能夠進入方格（35,37），因為3+5+3+7 = 18。但是，它不能進入方格（35,38），因為3+5+3+8 = 19。請問該機器人能夠達到多少個格子？

思路

回溯法，DFS

程式碼

int movx[] = {1,0,-1,0} ,movy[] = {0,1,0,-1};
int vis[100][100]={0};
int sum;
class Solution {
public:
    bool canreach(int k,int x,int y){
        int cnt = 0;
        while(x>0){
            cnt+=x%10;
            x=x/10;
        }
        while(y>0){
            cnt+=y%10;
            y=y/10;
        }
        return cnt<=k;
    }
    void dfs(int x,int y,int k,int m,int n)
    {
        for(int i=0;i<=3;++i)
        {
            int changed_x = x+movx[i],changed_y = y+movy[i];
            if(vis[changed_x][changed_y]==0&&changed_x>=0&&changed_x<m&&
              changed_y>=0&&changed_y<n&&canreach(k, changed_x,changed_y))
            {
                vis[changed_x][changed_y]=1;
                sum++;
                dfs(changed_x,changed_y,k,m,n);
            }
        }
    }

    int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<0)
            return 0;
        memset(vis,0,sizeof(vis));
        sum=1;
        vis[0][0]=1;
        dfs(0,0,threshold,rows,cols);
        return sum;
    }
};