66.机器人的运动范围——剑指offer

题目描述

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

/回溯法。仔细审题。。。
class Solution {
public:
    int movingCount(int threshold, int rows, int cols) {
        int res = 0;
        bool* flag = new bool[rows*cols];
        memset(flag, 0, rows*cols);
        dfs(threshold, rows, cols, flag, 0, 0, res);
        return res;
    }
    void dfs(int threshold, int rows, int cols, bool* flag, int i, int j, int &res) {
        if (i<0 || i >= rows || j<0 || j >= cols)
            return;
        if (*(flag + i*cols + j) == 1)
            return;
        if (check(i,j,threshold)){
            res++;
            *(flag + i*cols + j) = 1;
            dfs(threshold, rows, cols, flag, i, j - 1, res);//左
            dfs(threshold, rows, cols, flag, i, j + 1, res);//右
            dfs(threshold, rows, cols, flag, i - 1, j, res);//上
            dfs(threshold, rows, cols, flag, i + 1, j, res);//下
        }
    }
    bool check(int ii,int jj,int threshold){
        int sum = 0;
        while (ii != 0) {
            sum += ii % 10;
            ii = ii / 10;
        }
        while (jj != 0) {
            sum += jj % 10;
            jj = jj / 10;
        }
        return sum<=threshold;
    }
};