美文网首页
面试题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