【剑指offer】面试题13 - 机器人的运动范围

面试题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。请问该机器人能够到达多少个格子？

链接：LeetCode 牛客网

解法一：DFS + 回溯

class Solution {
    
    
public:
    int movingCount(int k, int rows, int cols) {
    
    
        if(k < 0 || rows <= 0 || cols <= 0) return 0;
        if(k == 0) return 1;
        bool* vis = new bool[rows * cols];
        memset(vis, 0, rows * cols);
        int count = DFS(k, rows, cols, 0, 0, vis);
        delete[] vis;
        return count;
    }
    
private:
    int DFS(int k, int rows, int cols, int row, int col, bool* vis)
    {
    
    
        int count = 0;
        if(row >= 0 && col >= 0 && row < rows && col < cols && sumBitNum(row) + sumBitNum(col) <= k && !vis[row * cols + col])
        {
    
    
            vis[row * cols + col] = true;
            count = 1 + DFS(k, rows, cols, row + 1, col, vis)
                      + DFS(k, rows, cols, row - 1, col, vis)
                      + DFS(k, rows, cols, row, col + 1, vis)
                      + DFS(k, rows, cols, row, col - 1, vis);
        }
        return count;
    }
    int sumBitNum(int num)
    {
    
    
        int sum = 0;
        while(num)
        {
    
    
            sum += num % 10;
            num /= 10;
        }
        return sum;
    }
};