机器人的运动范围 shuitang

题目

面试题13. 机器人的运动范围

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

解法

这题应该是典型的BFS题目,只是在扩展节点时加额外的限制条件即可。

代码如下:

class Solution {
    vector<int> dx = {-1,1,0,0};
    vector<int> dy = {0,0,-1,1};
public:
    int sumDigit(int x){
        int ans = 0;
        while(x){
            ans += x%10;
            x /= 10;
        }
        return ans;
    }
    bool isOK(int x, int y, int k){
        if(sumDigit(x)+sumDigit(y) > k) return false;
        return true;
    }
    int movingCount(int m, int n, int k) {
        
        queue<pair<int,int>> q;
        q.push(make_pair(0,0));
        pair<int,int> cur;
        unordered_map<int,int> mark;
        mark[0] = 1;
        int ans = 0;

        while(!q.empty()){
            cur = q.front();
            q.pop();
            ans++;

            int x = cur.first;
            int y = cur.second;
            
            for(int i=0;i<4;++i){
                int xx = x + dx[i];
                int yy = y + dy[i];
               
                if(xx <0 || yy <0 || xx>=m || yy>=n || !isOK(xx,yy,k) || mark.count(xx*n+yy) != 0) continue;
                q.push(make_pair(xx,yy));
                mark[xx*n+yy] = 1;
                
            }

        }
        return ans;
    }
};