【剑指 Offer】13. 机器人的运动范围

剑指 Offer 13. 机器人的运动范围

题目

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

思路

  本题使用广度优先搜索的方法进行解题，这里需要计算坐标的数位之和，那么需要对一个数先进行取余累加，再进行整除，直到该数字为0为止。

代码

class Solution {
    // 计算数位之和 先取余  在整除
    int get(int x)
    {
        int res = 0;
        while(x != 0)
        {
            res += x % 10;
            x = x / 10;
        }
        return res;
    }

public:
    int movingCount(int m, int n, int k) {
        if(k == 0) return 1;// 第一个格子

        queue<pair<int,int>> Q;

        // 偏移数组 向右或者向下
        int dx[2] = {0,1};
        int dy[2] = {1,0};

        vector<vector<int>> vis(m,vector<int>(n,0));
        Q.push(make_pair(0,0));
        vis[0][0] = 1;// 标记第一个数组
        int ans = 1;// 计数器

        while(!Q.empty())
        {
            auto[x,y] = Q.front();// 取出队列第一个元素
            Q.pop();// 去除该元素
            // 遍历两个方向
            for(int i = 0; i < 2; i++)
            {
                int tx = dx[i] + x;
                int ty = dy[i] + y;

                if(tx < 0 || tx >= m || ty < 0 || ty >= n || vis[tx][ty] || get(tx) + get(ty) > k)
                {
                    // 判断是否越界 并且判断该点是否已经访问过，判断下标的数位之和是否超过K
                    continue;// 直接寻找下一个方向
                }

                Q.push(make_pair(tx,ty));// 将改网格送入队列中
                vis[tx][ty] = 1;// 标记该网格已经被访问过
                ans++;// 计数器加加
            }
        }
        return ans;
    }
};