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

题目：

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

分析：

和前一题有些类似，将m×n的方格看成一个m×n的矩阵。除了边界之外，其他格子都有4个方向可以探测。

解法：

package com.wsy;

public class Main {
    public static void main(String[] args) {
        int rows = 10;
        int cols = 10;
        int k = 10;
        int result = getCount(rows, cols, k);
        System.out.println(result);
    }

    public static int getCount(int rows, int cols, int k) {
        if (rows < 1 || cols < 1 || k < 0) {
            return 0;
        }
        boolean[][] visit = new boolean[rows][cols];
        return getCountCore(rows, cols, 0, 0, k, visit);
    }

    public static int getCountCore(int rows, int cols, int row, int col, int k, boolean[][] visit) {
        int count = 0;
        if (row >= 0 && row < rows && col >= 0 && col < cols && !visit[row][col] && check(row, col, k)) {
            visit[row][col] = true;
            count = 1 + getCountCore(rows, cols, row + 1, col, k, visit) + getCountCore(rows, cols, row - 1, col, k, visit) + getCountCore(rows, cols, row, col + 1, k, visit) + getCountCore(rows, cols, row, col - 1, k, visit);
        }
        return count;
    }

    public static boolean check(int row, int col, int k) {
        int sum = 0;
        while (row != 0) {
            sum += row % 10;
            row /= 10;
        }
        while (col != 0) {
            sum += col % 10;
            col /= 10;
        }
        return sum <= k;
    }
}