机器人的运动范围
机器人的运动范围
http://www.nowcoder.com/questionTerminal/6e5207314b5241fb83f2329e89fdecc8
题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
【DFS】经典dfs: 先把不能走的格子标记为true
class Solution { public: int n,m,k; bool f[1000][1000]; int ans=0; int dir[4][2]={{0,1},{0,-1},{-1,0},{1,0}}; int compute(int x,int y) { int sum=0; while(x) { sum+=x%10; x/=10; } while(y) { sum+=y%10; y/=10; } return sum; } void dfs(int x,int y) { if(x<0||x>n-1||y<0||y>m-1||f[x][y]) return; f[x][y]=true; ans+=1; for(int i=0;i<4;i++) { int dx=x+dir[i][0]; int dy=y+dir[i][1]; dfs(dx,dy); } } int movingCount(int threshold, int rows, int cols) { n=rows; m=cols; k=threshold; memset(f,false,sizeof(f)); for(int i=0;i<n;i++) { for(int j=0;j<m;j++) { if(compute(i,j)>threshold) f[i][j]=true; } } dfs(0,0); return ans; } };