【剑指Offer】机器人运动

题目描述

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

思路分析

这个方格也可以看出一个m*n的矩阵。同样在这个矩阵中，除边界上的格子之外其他格子都有四个相邻的格子。
机器人从坐标(0,0)开始移动。当它准备进入坐标为(i,j)的格子时，通过检查坐标的数位和来判断机器人是否能够进入。如果机器人能够进入坐标为(i,j)的格子，我们接着再判断它能否进入四个相邻的格子(i,j-1)、(i-1,j),(i,j+1)和(i+1,j)。

代码实现

public class Solution {
    public  int movingCount(int threshold, int rows, int cols) {
        // 定义一个指示数组，判断当前位置是否已经访问过
        boolean[][] visited = new boolean[rows][cols];
        int movingCount = movingCountCore(threshold, rows, cols, 0, 0, visited);
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (visited[i][j]){
                    System.out.print(visited[i][j] + "  ");
                }else {
                    System.out.print(visited[i][j] + " ");
                }
            }
            System.out.println();
        }
        return movingCount;
    }

    /**
     * @param threshold
     * @param rows
     * @param cols
     * @param row
     * @param col
     * @return
     */
    private  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][col] = true;
            count += 1 + movingCountCore(threshold, rows, cols, row, col - 1, visited)/*左*/
                    + movingCountCore(threshold, rows, cols, row, col + 1, visited)/*右*/
                    + movingCountCore(threshold, rows, cols, row - 1, col, visited)/*上*/
                    + movingCountCore(threshold, rows, cols, row + 1, col, visited)/*下*/;
        }
        return count;
    }

    private boolean check(int threshold, int rows, int cols, int row, int col, boolean[][] visited) {
        if (row >= 0 && row < rows && col >= 0 && col < cols
                && positionSum(row, col) <= threshold
                && !visited[row][col]) {
            // 说明可以访问这个点
            return true;
        } else {
            return false;
        }
    }

    /**
     * 计算当前坐标各数位之和
     *
     * @param row
     * @param col
     * @return
     */
    private int positionSum(int row, int col) {
        int sum = 0;
        while (row != 0) {
            sum += row % 10;
            row /= 10;
        }
        while (col != 0) {
            sum += col % 10;
            col /= 10;
        }
        return sum;
    }
}