剑指offer--66.机器人的运动范围【完】

题目描述

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

时间限制：1秒 空间限制：32768K 热度指数：125433

思路

深搜，回溯法，详细见代码

class Solution {
    //深搜，回溯法
public:
    int movingCount(int k, int rows, int cols)
    {
        int count=0;    //count为能到达的格子数
        bool flag[rows*cols];//标记是否已经访问过，防止重复计数
        for(int i=0;i<rows;++i)
            for(int j=0;j<cols;++j)
                flag[i*cols+j]=false;
        movingCountMain(k, rows, cols,count,0,0,flag);
        return count;
    }
    void movingCountMain(int k, int rows, int cols,int &count,int i,int j,bool flag [])
    {//i表示当前行，j表示当前列
        
        //当越界，或者相加大于k时返回false
        if(j<0||i<0||i>=rows||j>=cols||(numToCount(i)+numToCount(j))>k)
            return ;
        //否则增加计数值，并向上下左右移动一格
        if(!flag[i*cols+j])
        {++count;
         flag[i*cols+j]=true;
        movingCountMain(k, rows, cols,count,i+1,j,flag);
        movingCountMain(k, rows, cols,count,i-1,j,flag);
        movingCountMain(k, rows, cols,count,i,j+1,flag);
        movingCountMain(k, rows, cols,count,i,j-1,flag);
        }

    }
    int numToCount(int num)
    {//将十进制数字的每一位相加
        int count=0;
        while(num>0)
        {
            count+=num%10;
            num=num/10;
        }
        return count;
    }
};