剑指offer 面试题13 机器人的运动范围

机器人的运动范围

题目

地上有一个m行n列的方格。一个机器人从坐标（0,0）的格子开始移动，它每次可以向左，向右，向上，向下移动一格，但不能进入行坐标和列坐标的位数之和大于k的格子。例如：当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18；但它不能进入方格（35,38），因为3 + 5+3+8 = 19.请问该机器人最多能到达多少个格子？

思路

机器人从坐标（0,0）开始移动。当它准备进入坐标为（i，j）的格子时通过检查坐标的位数和来判断机器人是否能够进入。如果机器人能够进入坐标为（i,j）的格子，则再判断它能否进入4个相邻的格子（i,j-1）、（i-1，j）、（i,j+1）和（i+1，j）

代码

public static int movingCount(int threshold, int rows, int cols){
	if(thresold<0||rows<=0||cols<=0){
		return 0;
	}
	boolean[] visited = new boolean[rows*cols];
	int count= movingCountCore(threshold, rows, cols, 0,0,visited);
	return visited;
}
public static int movingCountCore(int threshold, int rows, int cols, int row, int cols, boolean visited){
	int count=0;
	if(check(threshold,rows,cols,row,col,visited)){
		visited[riws*cols+col]=true;
		count=1+movinngCountCore(threshold,rows,cols,row-1,col,visited)+movinngCountCore(threshold,rows,cols,row,col-1,visited)+movinngCountCore(threshold,rows,cols,row+1,col,visited)+movinngCountCore(threshold,rows,cols,row,col+1,visited);
	}
	return count;

}
public boolean check(int threshold, int rows, int cols, int row, int col, boolean visited){
	if(row>=0&&row<rows&&col>=0&&col<cols&& getDigitSum(col)+getDigitSum(row)<=threshold&&!visited[rows*cols+row]){
		return true;
	}
	return false;
}
public int getDigitSum(int num){
	int sum=0;
	while(num>0){
		sum+=num%10;
		num/=10;
	}
	return sum;
}