剑指offerWeek2

周一：机器人的运动范围

题目链接：机器人的运动范围

地上有一个 m行和 n列的方格，横纵坐标范围分别是 0∼m−1和 0∼n−1
。一个机器人从坐标 (0,0)的格子开始移动，每一次只能向左，右，上，下四个方向移动一格。但是不能进入行坐标和列坐标的数位之和大于 k的格子。请问该机器人能够达到多少个格子？注意:0<=m<=50
0<=n<=50
0<=k<=100
样例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。

AC代码

class Solution {
public:queue<pair<int, int>> q;int getOne(int x){int s = 0;while (x) s += (x % 10), x /= 10;return s;}int getSum(auto p){return getOne(p.first) + getOne(p.second);}int movingCount(int threshold, int rows, int cols){if (!rows || !cols) return 0;q.push({0, 0});int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};int res = 0;vector<vector<bool>> st(rows, vector<bool>(cols));while (q.size()){auto t = q.front();q.pop();if (getSum(t) > threshold || st[t.first][t.second]) continue;res ++ ;st[t.first][t.second] = true;for (int i = 0; i < 4; i ++ ){int a = t.first + dx[i], b = t.second + dy[i];if (a >= 0 && b >= 0 && a < rows && b < cols) q.push({a, b});}}return res;}
};

思路：

整体思路

二维数组，统计元素特征，可以使用DFS也可以BFS，这里以BFS为例BFS步骤
- 创建队列
- 出发点入队
- while (队列有元素)- 取出队首元素- 判断是否满足某些条件，不满足则进入下次循环- 判断当前位置可以走的下一个位置的一些边界条件- 下一个位置加入队列