(C++版)劍指offer——機器人的運動範圍
阿新 • • 發佈:2019-01-12
題目描述
地上有一個m行和n列的方格。一個機器人從座標0,0的格子開始移動,每一次只能向左,右,上,下四個方向移動一格,但是不能進入行座標和列座標的數位之和大於k的格子。 例如,當k為18時,機器人能夠進入方格(35,37),因為3+5+3+7 = 18。但是,它不能進入方格(35,38),因為3+5+3+8 = 19。請問該機器人能夠達到多少個格子?
思路
回溯法,DFS
程式碼
int movx[] = {1,0,-1,0} ,movy[] = {0,1,0,-1}; int vis[100][100]={0}; int sum; class Solution { public: bool canreach(int k,int x,int y){ int cnt = 0; while(x>0){ cnt+=x%10; x=x/10; } while(y>0){ cnt+=y%10; y=y/10; } return cnt<=k; } void dfs(int x,int y,int k,int m,int n) { for(int i=0;i<=3;++i) { int changed_x = x+movx[i],changed_y = y+movy[i]; if(vis[changed_x][changed_y]==0&&changed_x>=0&&changed_x<m&& changed_y>=0&&changed_y<n&&canreach(k, changed_x,changed_y)) { vis[changed_x][changed_y]=1; sum++; dfs(changed_x,changed_y,k,m,n); } } } int movingCount(int threshold, int rows, int cols) { if(threshold<0) return 0; memset(vis,0,sizeof(vis)); sum=1; vis[0][0]=1; dfs(0,0,threshold,rows,cols); return sum; } };