剑指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)
    {
        if(threshold<0||rows<1||cols<1)
            return 0;
        int countnum=0;
        bool *visited=new bool[rows*cols];//用于标记已访问过的格子
        fill(visited,visited+cols*rows,false);
        travel(threshold,0,0,rows,cols,countnum,visited);
        delete[] visited;
        return countnum;
    }
    
    void travel(int threshold,int row,int col,int rows,int cols,int &countnum,bool* visited){
        if(count(row)+count(col)>threshold)
            return ;
        if(row>=0&&row<rows&&col>=0&&col<cols&&!visited[row*cols+col]){
            visited[row*cols+col]=true;
            countnum+=1;
            travel(threshold,row+1,col,rows,cols,countnum,visited);
            travel(threshold,row-1,col,rows,cols,countnum,visited);
            travel(threshold,row,col+1,rows,cols,countnum,visited);
            travel(threshold,row,col-1,rows,cols,countnum,visited);
        }
    }
    
    int count(int num){
        int sum=0;
        while(num>0){
            sum+=num%10;
            num/=10;
        }
        return sum;
    }
};