机器人的运动路径

机器人的运动路径

题目

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

思路

还是先定义一个从某点出发的函数，只要满足条件能够到达，则计数+1。与前面题目矩阵中的路径不同的是，此题求的是能够到达的点数，不是存在的路径，故上下左右用加法而不是或。

class Solution {
private:
    int movingCountCore(int i, int j, int m, int n, int k, vector<vector<bool>> &falg) {
        if (i < 0 || i >= m || j < 0 || j >= n || (i/10 + i%10 + j/10 + j%10) > k || falg[i][j] == true) {
            return 0;
        }

        falg[i][j] = true;
        return 1 + movingCountCore(i + 1, j, m, n, k, falg) +
                movingCountCore(i - 1, j, m, n, k, falg) +
                movingCountCore(i, j + 1, m, n, k, falg) +
                movingCountCore(i, j - 1, m, n, k, falg);
    }
public:
    int movingCount(int m, int n, int k) {
        if (m <=0 || n <=0 || k < 0) {
            return 0;
        }

        vector<vector<bool>> falg(m, vector<bool>(n, false));
        return movingCountCore(0, 0, m, n, k, falg);
    }
};