機器人的運動範圍

題目描述

地上有一個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 count=0；int numSum(int num)
    {
        int rev=0;
        while(num)
        {
            rev+=num%10;
            num=num/10;
        }
        return rev;
    }
    void fun(int x,int y,int threshold,int rows,int cols,bool * flags)
    {
        if(x<0||y<0||x>=rows||y>=cols||flags[x*cols+y])
            return;
        if(flags[x*cols+y]==false&&(numSum(x)+numSum(y)<=threshold) )
        {
            flags[x*cols+y]=true;
            count++;
            fun(x+1,y,threshold,rows,cols,flags);
            fun(x-1,y,threshold,rows,cols,flags);
            fun(x,y+1,threshold,rows,cols,flags);
            fun(x,y-1,threshold,rows,cols,flags);
        }
    }
    int movingCount(int threshold, int rows, int cols)
    {
        bool * flags=new bool[rows*cols];
        for(int i=0;i<rows;i++)
        {
            for(int j=0;j<cols;j++)
                flags[i*cols+j]=false;
        }
        fun(0,0,threshold,rows,cols,flags);
        return count;
    }
};