剑指offer(java版)面试题13:机器人的运动范围

题目描述

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

思路:

1.使用二维数组记录方格信息,true:机器人能走到;false:机器人走不到

2.从座标[0,0]开始往下走,如果当前座标符合要求,数组置为true,如果数组已经为true,说明已经走过,为了防止循环递归,数组为true时,返回。

3.递归调用附近四个座标


代码:

public class Solution {
    public static int movingCount(int threshold, int rows, int cols) {
        boolean[][] path = new boolean[rows][cols];
        movingCount(path, rows, cols, 0, 0, threshold);
        int num = 0;
        for (boolean[] p : path) {
            for (boolean b : p) {
                if (b) {
                    num++;
                }
            }
        }
        return num;
    }

    public static void movingCount(boolean[][] path, int rows, int cols, int x, int y, int result) {
        if (checkPath(path, rows, cols, x, y, result)) {
            path[x][y] = true;
            movingCount(path, rows, cols, x - 1, y, result);
            movingCount(path, rows, cols, x + 1, y, result);
            movingCount(path, rows, cols, x, y - 1, result);
            movingCount(path, rows, cols, x, y + 1, result);
        }
    }

    public static boolean checkPath(boolean[][] path, int rows, int cols, int x, int y, int result) {
        if (x < 0 || x >= rows || y < 0 || y >= cols) {
            return false;
        }
        if (path[x][y]) {
            return false;
        }
        int k = 0;
        while (x != 0) {
            k += x % 10;
            x = x / 10;
        }

        while (y != 0) {
            k += y % 10;
            y = y / 10;
        }
        return k <= result;
    }
}

点赞