LeetCode:机器人的运动范围

地上有一个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。请问该机器人能够到达多少个格子?

示例 1:

输入:m = 2, n = 3, k = 1
输出:3

示例 2:

输入:m = 3, n = 1, k = 0
输出:1

提示:

1 <= n,m <= 100
0 <= k <= 20

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof

思路:

本题是求机器人的运动范围,即求机器人所能达到的区域,是否能联想到广度搜索,从一个点散开来搜索,形象类似于一滴水引起的涟漪,从一个点往四周搜索到符合的点放入队列,再从队列依次取出去搜索它四周符合的点,再放入队列,这样不就完成了广度搜索了嘛,这里需要注意的是访问过得点无需再访问了,我们访问过了可以给他标记一下。


代码题解:

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
int getNumSum(int n, int total) { //得到一个数字的每个位上的数字之和
    if (n/10 == 0 )
    {
        return total + n;
    }
    return getNumSum(n / 10, total + n % 10);
}
int movingCount(int m, int n, int k) {
    if (k == 0) 
    {
        return 1;
    }
    vector<vector<int>> vis(m,vector<int>(n)); //已经访问过的点了
    queue <pair<int, int>> Q; //待扩展开来的队列

    int dx[2] = { 0,1 }; // x 方向 不向右或者向右
    int dy[2] = { 1,0 }; //y 方向 向下或者不向下 ,和dx合起来就是向右或者向下搜索

    Q.push(make_pair(0, 0)); //将第一个点放入
    vis[0][0] = 1; //坐标为[0,0]标记已访问
    int sum = 1; //起始范围为1
    while (!Q.empty()) //如果队列不为空代表可继续搜索
    {
        pair<int,int> member = Q.front(); //队首的点赋值
        Q.pop();//弹出队首
        for (int i = 0; i < 2; i++) //遍历两个方向
        {
            int x = member.first + dx[i];
            int y = member.second + dy[i];
            if (x<0 || y<0 || x>=m || y>=n ||((getNumSum(x,0)+ getNumSum(y,0))>k)||(vis[x][y])) //不符合条件,如果访问越界或者2坐标位数相加大于k或者已经访问过了continue
            {
                continue;
            }
            //符合条件
            Q.push(make_pair(x, y)); //加入队列
            sum++; //范围+1
            vis[x][y] = 1; //标记已访问
        }
    }
    return sum;
}
int main() {
    cout << movingCount(2, 3, 1) << endl;
    cout << movingCount(3, 1, 0) << endl;
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容