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

示例 1：

输入：m = 2, n = 3, k = 1
输出：3
示例 2：

输入：m = 3, n = 1, k = 0
输出：1
提示：
* 1 <= n,m <= 100
* 0 <= k <= 20

思路：回溯法

（1）设置visited数组记录坐标访问情况
（2）从[0][0]开始递归，许可访问的条件是它的行坐标列坐标之和大于k。

边界检查：

（1）机器人不得越过矩阵界限

代码：

class Solution {
    public:
        int rst = 0;
        int movingCount(int m, int n, int k) {
            vector<vector<bool>> visited(m, vector<bool>(n, false));


            moving(m, n, 0, 0, k, visited);
            return rst;
        }


        void moving(int rows, int cols, int row, int col, int k, vector<vector<bool>>& visited) {
            //越界处理
            if (row >= 0 && row < rows && col >= 0 && col < cols
                && digitsSum(row) + digitsSum(col) <= k && !visited[row][col]) {
                ++rst;
                visited[row][col] = true;


                moving(rows, cols, row + 1, col, k, visited);
                moving(rows, cols, row - 1, col, k, visited);
                moving(rows, cols, row, col + 1, k, visited);
                moving(rows, cols, row, col - 1, k, visited);
            }


            return;
        }
        
        int digitsSum(int num) {
            int sum = 0;
            while (num != 0) {
                sum += num % 10;
                num /= 10;
            }
            return sum;
        }
    };