剑指offer:机器人的运动范围

描述

地上有一个 rows 行和 cols 列的方格。坐标从 [0,0] 到 [rows-1,cols-1] 。一个机器人从坐标 [0,0] 的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于 threshold 的格子。 例如，当 threshold 为 18 时，机器人能够进入方格   [35,37] ，因为 3+5+3+7 = 18。但是，它不能进入方格 [35,38] ，因为 3+5+3+8 = 19 。请问该机器人能够达到多少个格子？

数据范围： 0 \le threshold \le 15 \0≤threshold≤15  ，1 \le rows,cols \le 100 \1≤rows,cols≤100 

进阶：空间复杂度 O(nm) \O(nm)  ，时间复杂度 O(nm) \O(nm) 

示例1

输入：

1,2,3

复制返回值：

3

思路:

深度优先遍历

class Solution {
public:
    int col,row;
    int threshold;
    int res=0;
    //方向
    int dx[4]={-1,0,1,0};
    int dy[4]={0,1,0,-1};
    
    int check(int n){
        int sum=0;
        while(n){
            sum+= (n%10);
            n /=10;
        }
        return sum;
    }
    
    void dfs(int x,int y,vector<vector<int>> &mark){
        //检查下标
        if(x<0 || x>=row || y<0 || y>=col || mark[x][y]==1) return;
        //检查下标是否满足条件
        if(check(x)+check(y)>threshold) return;
        mark[x][y]=1;
        res+=1;
        //上下左右深度遍历
        for(int i=0;i<4;i++){
            dfs(x+dx[i],y+dy[i],mark);
        }
    }
    
    int movingCount(int threshold, int rows, int cols) {
        if(threshold<0) return 0;
        col=cols,row=rows;
        this->threshold=threshold;
        vector<vector<int>> mark(row,vector<int>(col,-1));
        dfs(0,0,mark);
        return res;
    }
};