13. 机器人的运动范围

题目描述：

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

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

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

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

样例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

算法1：

BFS

class Solution {
public:
    int movingCount(int threshold, int rows, int cols)
    {
        if( rows <=0 || cols <=0 || threshold < 0)
            return 0;
        int ans = 0;
        //int visit[rows][cols];
        //memset(visit, 0, sizeof(visit));
        vector<vector<int>>visit(rows, vector<int>(cols, 0));
        queue<pair<int, int>>Q;
        Q.push(make_pair(0,0));
        visit[0][0] = 1;
        int dx[] = {0,1,0,-1};
        int dy[] = {1,0,-1,0};
        while(!Q.empty())
        {
            int x = Q.front().first;
            int y = Q.front().second;
            
            ans++;
            Q.pop();
            for(int i=0;i<=3;i++)
            {
                int x1 = x+dx[i];
                int y1 = y+dy[i];
                if(x1 >=0 && x1<rows && y1>=0 && y1<cols && check(threshold, x1, y1) && !visit[x1][y1])
                {
                    Q.push(make_pair(x1, y1));
                    visit[x1][y1] = 1;
                }
                
            }
        }
        return ans;
    }
    
    
    bool check(int threshold, int x, int y)
    {
        int val = 0;
        while(x)
        {
            val += x%10;
            x /= 10;
        }
        while(y)
        {
            val+=y%10;
            y/=10;
        }
        if(val <= threshold)
            return true;
        else
            return false;
    }
};

算法2：

DFS

class Solution {
public:
    
    int movingCount(int threshold, int rows, int cols)
    {
        if(rows <=0 || cols <= 0 || threshold<0)
            return 0;
        vector<vector<int>>visit(rows, vector<int>(cols, 0));
        int ans = 1;
        visit[0][0] = 1;
        dfs(0,0, visit,threshold, rows, cols, ans);
        return ans;
    }
    
    void dfs(int x, int y, vector<vector<int>>& visit, int & threshold, int &rows, int & cols, int &ans)
    {
        int dx[4]={0, 1, 0, -1};
        int dy[4]={1, 0, -1, 0};
        
        visit[x][y] = 1;
        for(int i=0;i<=3;i++)
        {
            int nx = x+dx[i], ny=y+dy[i];
            if(nx>=0 && nx<rows && ny>=0 && ny<cols && check(threshold, nx, ny) && !visit[nx][ny])
            {
                
                ans++;
                dfs(nx, ny, visit, threshold, rows, cols, ans);
            }
        }
    }
    
    bool check(int threshold, int x, int y)
    {
        int val=0;
        while(x)
        {
            val += x%10;
            x /= 10;
        }
        while(y)
        {
            val += y%10;
            y /= 10;
        }
        return threshold >= val? true: false;
    }
};