刷题22-机器人的运动范围

原题链接

题目描述

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

参考解法(回溯法遍历整个数组)

// 这里不能直接用两个for循环进行遍历，因为一次只能移动一格
// 所以采用回溯法进行遍历
public class test {
     
	public int movingCount(int k, int rows, int cols) {
     
		// 记录是否已经走过
		int flag[][] = new int[rows][cols];
		// 从(0,0)出发，走过的标记为1，未走过的为0
		return helper(0, 0, rows, cols, flag, k);
	}

	public int helper(int i, int j, int rows, int cols, int[][] flag, int k) {
     
		// 探索时，判断当前节点是否可达的标准为：
		// 1）当前节点在矩阵内；
		// 2）当前节点未被访问过；
		// 3）当前节点满足limit限制。
		if (i < 0 || i >= rows || j < 0 || j >= cols || numSum(i) + numSum(j) > k || flag[i][j] == 1)
			return 0;
		// 表示当前节点已经访问，标记为1
		flag[i][j] = 1;
		// 然后从当前位置往四个方向探索，
		// 返回1 + 4 个方向的探索值之和。
		return helper(i - 1, j, rows, cols, flag, k) 
				+ helper(i + 1, j, rows, cols, flag, k)
				+ helper(i, j - 1, rows, cols, flag, k) 
				+ helper(i, j + 1, rows, cols, flag, k) 
				+ 1;
	}

	public int numSum(int i) {
     
		int sum = 0;
		do {
     
			sum += i % 10;
		} while ((i = i / 10) > 0);
		return sum;
	}
}