面试题67：机器人的运动范围

最新推荐文章于 2022-09-09 22:43:00 发布

原创最新推荐文章于 2022-09-09 22:43:00 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种通过回溯算法解决机器人在限定条件下的可行走路径数量问题的方法。机器人从起点出发，在不超过阈值k的行与列数位和限制下，计算能够到达的有效格子数目。

题目：地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

这一题比上一题容易，用变相的回溯，不用回到原来的点。代码如下：

class Solution {
public:
    int movingCount(int threshold, int rows, int cols)
    {
        if(threshold <= 0)
            return 0;
        bool *visited = new bool[rows*cols];
        memset(visited, 0, rows*cols);
        int num = 0;
        moving(threshold, rows, cols, 0, 0, num, visited);
        return num;
        
    }
    void moving(int threshold, int rows, int cols, int row, int col, int& num, bool* visited)
    {
        if(sum(row, col) > threshold)
            return;
        if(row >=0 && row < rows && col >=0 && col < cols && (!visited[row*cols + col]))
        {
            visited[row*cols+col] = true;
            num++;
            moving(threshold, rows, cols, row+1, col, num, visited);
            moving(threshold, rows, cols, row-1, col, num, visited);
            moving(threshold, rows, cols, row, col+1, num, visited);
            moving(threshold, rows, cols, row, col-1, num, visited);
        }
        return;
    }
    int sum(int row, int col)
    {
        int num = 0;
        while(row != 0)
        {
            num +=(row % 10);
            row /= 10;
        }
        while(col != 0)
        {
            num +=(col % 10);
            col /= 10;
        }
        return num;
    }
};