機器人的運動範圍

牛客上看到這樣一道題:

題目描述

地上有一個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)
    {
        bool *visited=new bool[rows*cols];  
    	memset(visited,0,rows*cols);
        int count = 0;
        moveHelp(threshold,rows,cols,0,0,visited,count);
        return count;
    }
    
    void moveHelp(int threshold,int rows,int cols,int row,int col,bool* visited,int& count){
        if(row>=0&&row<rows
          &&col>=0&&col<cols
          &&!visited[cols*row+col]
          &&util(row)+util(col)<=threshold){
            count++;
            visited[cols*row+col] = true;
            moveHelp(threshold,rows,cols,row+1,col,visited,count);
            moveHelp(threshold,rows,cols,row-1,col,visited,count);
            moveHelp(threshold,rows,cols,row,col+1,visited,count);
            moveHelp(threshold,rows,cols,row,col-1,visited,count);       
        }
        
        
    }
    
    int util(int num){
        int sum = 0;
        while(num != 0){
         	sum += num%10;
            num /= 10;
        }
        return sum;
        
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章