美文网首页
面试题13:机器人的运动范围

面试题13:机器人的运动范围

作者: 修司敦 | 来源:发表于2018-11-14 10:37 被阅读0次
地上有一个m行n列的方格。一个机器人从坐标(0, 0)的格子开始移动,它每一次可以向左、右、上、下移动一格,但不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格(35, 37),因为3+5+3+7=18。但它不能进入方格(35, 38),因为3+5+3+8=19。请问该机器人能够到达多少个格子?

解析:
该题明显使用回溯法。目标就是从 (0,0) 这个点开始走,遍历完所有能走到的方格。每个格子都可能有上、下、左、右四个方向中某几个无法前进,那么遇到无法前进的就不继续走。每次遇到一个没有被访问过的方格,而且这个方格可以走,那就立马把这个格子设置成已访问,同时把计数器加一。接下来从其他方向访问,遇到了已经访问过了的方格,那直接就不去访问就好了。思路确定了之后,想一下要实现的几个函数:计算出当前方格的那个“数位之和”的函数、检查当前方格能不能进入的函数、回溯的核心函数、总的调用函数。回溯法一般使用递归的实现,在这个题就是,遇到了能访问的就在这个方格上继续往四周探索,不能的话就立马终止。


答案:

int getDigitSum(int number)
{
    int cnt = 0;
    while (number>0)
    {
        cnt += number%10;
        number /= 10;
    }
    return cnt;
}

inline bool check(int threshold, unsigned int rows, unsigned int cols, \
            unsigned int row, unsigned int col, const bool* visited)
{
    return row<rows && col<cols && row>=0 && col>=0 && \
    !visited[row*cols+col] && threshold>=(getDigitSum(row)+getDigitSum(col));
}

int movingCountCore(int threshold, unsigned int rows, unsigned int cols, \
                unsigned int row, unsigned int col, bool* visited)
{
    int cnt = 0;
    if (check(threshold, rows, cols, row, col, visited))
    {
        visited[row * cols + col] = true;
        cnt = 1+movingCountCore(threshold, rows, cols, row, col+1, visited)+ \
                movingCountCore(threshold, rows, cols, row, col-1, visited)+ \
                movingCountCore(threshold, rows, cols, row+1, col, visited)+ \
                movingCountCore(threshold, rows, cols, row-1, col, visited);
    }
    return cnt;
}

int movingCount(int threshold, unsigned int rows, unsigned int cols)
{
    if(threshold<0 || rows<=0 || cols<=0) return 0;
    bool *visited = new bool[rows*cols];
    memset(visited, false, rows*cols);

    int count = movingCountCore(threshold, rows, cols, 0, 0, visited);

    delete[] visited;
    return count;
}

相关文章

网友评论

      本文标题:面试题13:机器人的运动范围

      本文链接:https://www.haomeiwen.com/subject/bhcdfqtx.html