剑指offer 面试题13 机器人的运动范围

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

tips: bfs，记得设置isVisited矩阵，否则可能会复杂度过大。
书上也有用递归做的，代码比较简洁。

class Solution {
public:
    bool isVaildPos(int threshold, int x, int y) {  
            int sum = 0;
            while (x > 0) {
                sum += x % 10;
                x /= 10;
            }
            while (y > 0) {
                sum += y % 10;
                y /= 10;
            }
            return sum <= threshold;
    }

    int movingCount(int threshold, int rows, int cols)
    {
        if(rows<=0||cols<=0||threshold<0) {
            return 0;
        }
        vector<vector<bool>> isVisited=vector<vector<bool>>(rows,vector<bool>(cols,false)); 
        queue<pair<int,int>> que;
        if (isVaildPos(threshold,0,0))
        {
            isVisited[0][0]=true;
            que.push({0,0});
        }
        vector<pair<int,int>> move={{-1,0},{0,1},{1,0},{0,-1}};
        int count=0;
        while(!que.empty()) {
            pair<int,int> pos=que.front();
            que.pop();
            
            count++;
            for (int i = 0; i < move.size(); i++)
            {
                int nextPosX=pos.first+move[i].first;
                int nextPosY=pos.second+move[i].second;
                if(nextPosX<0||nextPosX>=rows||nextPosY<0||nextPosY>=cols) {
                    continue;
                }
                if(isVisited[nextPosX][nextPosY]==false&&isVaildPos(threshold,nextPosX,nextPosY)) {
                    isVisited[nextPosX][nextPosY]=true;
                    que.push({nextPosX,nextPosY});
                }
            }
        }
        return count;
    }
};

来源：CSDN

作者：wenkun97

链接：https://blog.csdn.net/hello_my_coder/article/details/103947333