剑指Offer面试题13：机器人的运动范围 Java实现

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

原题描述：

面试题13：机器人的运动范围

地上有一个m行n列的方格，机器人从坐标（0,0）的各自开始移动，每次可以往上下左右移动一格，但不能进入行列坐标的数位之和大于k的格子。问机器人能够达到多少个格子。

这题还是用回溯法解决问题，直接上Java代码了！

/**
	 * @param threshold 行列坐标的数位之和限制
	 * @param rows 方格行数
	 * @param cols 方格列数
	 * @return 能够到达的格子数
	 */
	public static int movingCount(int threshold, int rows, int cols) {
		if (threshold < 0 || rows <= 0 || cols <= 0) {
			return 0;
		}

		boolean[] visited = new boolean[rows * cols];
		int count = movingCountCore(threshold, rows, cols, 0, 0, visited);
		return count;
	}

/**
 * @param threshold
 * @param rows
 * @param cols
 * @param row 准备进入的格子行坐标
 * @param col 准备进入的格子列坐标
 * @param visited 标记每个格子是否已经记录过
 * @return
 */
	private static int movingCountCore(int threshold, int rows, int cols, int row, int col, boolean[] visited) {
		int count = 0;
		if (check(threshold, rows, cols, row, col, visited)) {
			visited[row * cols + col] = true;

			count = 1 + movingCountCore(threshold, rows, cols, row - 1, col, visited)
					+ movingCountCore(threshold, rows, cols, row, col - 1, visited)
					+ movingCountCore(threshold, rows, cols, row + 1, col, visited)
					+ movingCountCore(threshold, rows, cols, row, col + 1, visited);
		}
		return count;
	}

	//判断这个方格是否满足访问条件
	private static boolean check(int threshold, int rows, int cols, int row, int col, boolean[] visited) {
		if (row >= 0 && row < rows && col >= 0 && col < cols && getDigitSum(row) + getDigitSum(col) <= threshold
				&& !visited[row * cols + col])
			return true;

		return false;
	}

//计算下标各位数和
	private static int getDigitSum(int number) {
		int sum = 0;
		while (number > 0) {
			sum += number % 10;
			number /= 10;
		}
		return sum;
	}

测试代码如下：

public static void main(String[] args) {
		int [] kArr={0,1,2,3,4};
		int [] result=new int[kArr.length];
		for(int i=0;i<kArr.length;i++){
			result[i]=movingCount(kArr[i],3,6);
		}
		String s=Arrays.toString(result);
		System.out.println(s);
	}

输出结果为：[1, 3, 6, 9, 12]

做了Offer上两道回溯法的题目，但对这个概念还是有点蒙。立个flag，这个周末专门捋一下这个思想。