《剑指 Offer》——66、机器人的运动范围

1. 本题知识点

深度优先搜索法

2. 题目描述

地上有一个 m 行和 n 列的方格。一个机器人从坐标 0,0 的格子开始移动，每一次只能向左、右、上、下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于 k 的格子。

例如，当 k 为 18 时，机器人能够进入方格（35,37），因为 3+5+3+7 = 18。但是，它不能进入方格（35,38），因为 3+5+3+8 = 19。请问该机器人能够达到多少个格子？

3. 解题思路

使用深度优先搜索法进行求解，当搜索到某一格子时，若满足条件，继续向四个方向搜索；若不满足条件，则退回到上一个格子，重新选择要搜索的格子。（注意：已经搜索过的格子要做个标记）

注意：回溯是深度优先搜索的一种特例，它需要在每次搜索结束后清除标记状态，而深度优先搜索不需要清除标记状态。

4.代码

public class Solution {
    /**
     * 返回机器人所能到达的格子数
     *
     * @param threshold 位数之和不能大于 threshold
     * @param rows      行数
     * @param cols      列数
     * @return
     */
    public int movingCount(int threshold, int rows, int cols) {
        // 机器人所能到达的格子数
        int count = 0;
        // 创建标记数组，用来记录已经到达的格子数
        boolean[][] marked = new boolean[rows][cols];
        // 从方格的 0,0 出发
        dfs(0, 0, marked, threshold);
        // 遍历标记数组，得到机器人所能到达的格子数
        for (int i = 0; i < marked.length; i++) {
            for (int j = 0; j < marked[0].length; j++) {
                if (marked[i][j]) {
                    count++;
                }
            }
        }
        return count;
    }

    /**
     * 如果出发点能被到达，则标记出发点，并继续向其他方向访问
     * 否则返回上一个格子
     *
     * @param i         出发点的行下标
     * @param j         出发点的列下标
     * @param marked    标记数组
     * @param threshold 位数之和不能大于 threshold
     */
    private void dfs(int i, int j, boolean[][] marked, int threshold) {
        // 如果出发点在方格以外，则返回
        if (i < 0 || i > marked.length - 1 || j < 0 || j > marked[0].length - 1) {
            return;
        }
        // 如果出发点被标记了，则返回
        if (marked[i][j]) {
            return;
        }
        // 如果出发点位数之和大于 threshold，则返回
        if (sumOfDigit(i, j) > threshold) {
            return;
        }
        // 标记出发点
        marked[i][j] = true;
        // 继续向其他方向访问
        dfs(i - 1, j, marked, threshold);
        dfs(i + 1, j, marked, threshold);
        dfs(i, j - 1, marked, threshold);
        dfs(i, j + 1, marked, threshold);
        return;
    }

    /**
     * 返回 i 和 j 的位数之和
     *
     * @param i
     * @param j
     * @return
     */
    private int sumOfDigit(int i, int j) {
        int sum = 0;
        while (i > 0) {
            sum += i % 10;
            i = i / 10;
        }
        while (j > 0) {
            sum += j % 10;
            j = j / 10;
        }
        return sum;
    }
}