劍指Offer:機器人的運動範圍(13)
阿新 • • 發佈:2020-08-14
題目描述:
地上有一個m行和n列的方格。一個機器人從座標0,0的格子開始移動,每一次只能向左,右,上,下四個方向移動一格,但是不能進入行座標和列座標的數位之和大於k的格子。 例如,當k為18時,機器人能夠進入方格(35,37),因為3+5+3+7 = 18。但是,它不能進入方格(35,38),因為3+5+3+8 = 19。請問該機器人能夠達到多少個格子?
解題思路:
利用遞迴實現,每次只能走上下左右四個點,進行判斷點的位置是否越界,點數之和是否大於K,是否已經走過了。
1 public class Solution { 2 public int movingCount(int threshold, introws, int cols) 3 { 4 int flag[][]=new int[rows][cols];//記錄是否走過 5 return helper(0,0,rows,cols,flag,threshold); 6 } 7 8 private int helper(int i,int j,int rows,int cols,int [][] flag,int threshold){ 9 if(i<0 || i>=rows || j<0 || j>=cols || numSum(i)+numSum(j)> threshold || flag[i][j]==1){10 return 0; 11 } 12 flag[i][j]=1;//走過的位置設定為1 13 return helper(i-1,j,rows,cols,flag,threshold) 14 +helper(i+1,j,rows,cols,flag,threshold) 15 +helper(i,j-1,rows,cols,flag,threshold) 16 +helper(i,j+1,rows,cols,flag,threshold)+1; 17 }18 19 //行座標和列座標的數位之和 20 private int numSum(int i){ 21 int sum=0; 22 while(i>0){ 23 sum += i % 10; 24 i = i / 10; 25 } 26 return sum; 27 }