剑指offer题解合集——Week2day1

剑指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 (队列有元素)
	- 取出队首元素
	- 判断是否满足某些条件，不满足则进入下次循环
	- 判断当前位置可以走的下一个位置的一些边界条件
	- 下一个位置加入队列