CC BY 4.0 （除特别声明或转载文章外）
如果这篇博客帮助到你，可以请我喝一杯咖啡~
题目
面试题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;
}
};