剑指offer-机器人的运动范围

题目
地上有一个m行n列的方格。一个机器人从坐标(0, 0)的格子开始移动,它每一次可以向左、右、上、下移动一格,但不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格(35, 37),因为3+5+3+7=18。但它不能进入方格(35, 38),因为3+5+3+8=19。请问该机器人能够到达多少个格子?
思路
回溯法,走出一步,就看能不能走四周的格子。
代码

int MovingCount(int treshold,int rows,int cols)
{
    if(treshold<0||rows<0||cols<0)
    {
        return 0;
    }
    int*visited=new int[rows*cols];
    for(int i=0;i<rows*cols;i++)
        visited[i]=0;
    int row=0;
    int col=0;
    int count=MovingCountCore(treshold,rows,cols,row,col,visited);
    delete[]visited;
    return count;
}
int MovingCountCore(int treshold,int rows,int cols,int row,int col,int*visited)
{
    int count=0;
    if(check(treshold,rows,cols,row,col,visited))

    {
        visited[row*cols+col]=1;
        count=1+MovingCountCore(treshold,rows-1,cols,row,col,visited)+
        MovingCountCore(treshold,rows,cols,row,col-1,visited)+
        MovingCountCore(treshold,rows,cols,row+1,col,visited)+
        MovingCountCore(treshold,rows,cols,row,col+1,visited);
    }
}
bool check(int treshold,int rows,int cols,int row,int col,int*visited)
{
    int rownum=getdigit(row);
    int colnum=getdigit(col);
    if(treshold>=(rownum+colnum)&&row>=0&&row<rows&&col>=0&&col<cols&&!visited[row*cols+col])
        return true;
    else
         return false;
}
int getdigit(int k)
{
    int sum=0;
    while(!k)
    {
        sum+=(k%10);
        k=k/10
    }
    return sum;
}

猜你喜欢

转载自blog.csdn.net/gary_god/article/details/79965470