[剑指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);
    }
};

[剑指offer]机器人的运动范围[剑指offer]机器人的运动范围 probeTS 发布了10 篇原创文章 · 获赞 0 · 访问量 253 私信 关注
上一篇:剑指Offer:机器人的运动范围(Python语言实现)


下一篇:opencv学习笔记11-阈值操作threshold