剑指offer（Java实现）13 - 机器人的运动范围

机器人的运动范围-13

地上有一个 m 行和 n 列的方格，横纵坐标范围分别是 0 ~ m−1 和 0 ~ n−1。

一个机器人从坐标（0，0）的格子开始移动，每一次只能向左，右，上，下四个方向移动一格。

但是不能进入行坐标和列坐标的数位之和大于 k 的格子。

请问该机器人能够达到多少个格子？

样例1

输入：k=7, m=4, n=5
输出：20

样例2

输入：k=18, m=40, n=40
输出：1484
解释：当 k 为 18 时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。
      但是，它不能进入方格（35,38），因为3+5+3+8 = 19。

思路：

DFS深度优先搜索

public class Solution {
	 /**
     * 返回机器人能到达的格子数
     * @param threshold 阈值
     * @param rows 矩阵行数
     * @param cols 矩阵列数
     * @return
     */
	public int movingCount(int threshold, int rows, int cols) {
		
		if(threshold < 0 || rows <= 0 || cols <= 0)
			return 0;
		
		//初始化访问标识数组
		boolean[] visited = new boolean[rows * cols];

		return movingCountCore(threshold, rows, cols, 0, 0, visited);
	}

	/**
	 * 判断从第 row行,第 col列出发可以到达的格子数
	 */
	public 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;
	}
	/**
	 * 检查机器人能否到达该格子
	 */
	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(row) + getDigitSum(col) <= threshold
		        && !visited[row * cols + col])
			return true;
		
		return false;
	}
	/**
	 * 求一个数的各位数和
	 */
	public int getDigitSum(int num) {
		int sum = 0;
		while(num > 0) {
			sum += num % 10;
			num /= 10;
		}
		return sum;
    }
}