[剑指offer]机器人的运动范围

机器人的运动范围

题目

地上有一个 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。

题解

本题利用宽度优先搜索遍历，在所给矩阵中，依次遍历每个坐标，并判断坐标是否符合题目要求

注意：

• 机器人所能到达的位置均为连续的

class Solution {
public:
    int get_sum(int n) {
        int sum = 0;
        while (n != 0) {
            sum += n % 10;
            n /= 10;
        }
        return sum;
    }

    int bfs(int threshold, int rows, int cols) {
        int dx[]={0, 1, 0, -1}, dy[]={1, 0, -1, 0}, sum = 0;
        vector<vector<bool> > g(rows, vector<bool>(cols, false));
        queue<pair<int, int> > q;
        q.push({0, 0});
        g[0][0] = true;
        
        while (q.size()) {
            auto t = q.front();
            q.pop();
            int x = t.first, y = t.second;
            int tmp = get_sum(x) + get_sum(y);
            
            if (tmp > threshold) continue;
            sum ++;
            for (int i = 0; i < 4; i ++) {
                int a = x + dx[i], b = y + dy[i];
                if (a >= 0 && b >= 0 && a < rows && b < cols && !g[a][b]) {
                    q.push({a, b});
                    g[a][b] = true;
                }
            }
        }
        return sum;
    }

    int movingCount(int threshold, int rows, int cols) {
        if (!rows || ! cols) return 0;
        return bfs(threshold, rows, cols);
    }
};