题目:

较难  通过率:24.82%  时间限制:1秒  空间限制:64M

知识点递归

描述

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

数据范围: 0≤threshold≤15  ,1≤rows,cols≤100 

进阶:空间复杂度 O(nm) ,时间复杂度 O(nm)

示例1

输入:

1,2,3

返回值:

3

示例2

输入:

0,1,3

返回值:

1

示例3

输入:

10,1,100

返回值:

29

说明:

[0,0],[0,1],[0,2],[0,3],[0,4],[0,5],[0,6],[0,7],[0,8],[0,9],[0,10],[0,11],[0,12],[0,13],[0,14],[0,15],[0,16],[0,17],[0,18],[0,19],[0,20],[0,21],[0,22],[0,23],[0,24],[0,25],[0,26],[0,27],[0,28] 这29种,后面的[0,29],[0,30]以及[0,31]等等是无法到达的      

示例4

输入:

5,10,10

返回值:

21

代码:

class Solution {
public:
    int movingCount(int threshold, int rows, int cols) {
        if(rows<=0 && cols <= 0) return 0;
        
        int visited[rows][cols];
        //初始化
        for(int i= 0; i<rows;i++)
            for(int j=0;j<cols;j++)
                visited[i][j] = 0;
        return countMax(threshold, rows, cols, 0, 0, &visited[0][0]);
    }
    
   int countMax(int threshold, int rows, int cols, int r, int c, int* visited){
        if(r < 0 || c < 0|| r >=rows|| c>= cols || *(visited+r*cols+c) == 1) return 0;
        if(!checkCondition(r, c, threshold)) return 0;
       
        *(visited+r*cols+c)  = 1;
        return 1 + countMax(threshold,  rows, cols, r-1, c, visited) 
            + countMax(threshold,  rows, cols, r+1, c, visited) 
            + countMax(threshold,  rows, cols, r, c-1, visited) 
            + countMax(threshold,  rows, cols, r, c+1, visited) ;
    }
    
    bool checkCondition(int row, int col, int threshold){
        int sum = 0;
        while(row){
            sum += row %10;
            row /= 10;
        }
        while(col){
            sum += col %10;
            col /= 10;
        }
        return sum <= threshold;
    }
    
    
};