剑指offer13 机器人的运动范围

问题描述

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

代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int m = scanner.nextInt();
        int n = scanner.nextInt();
        int k = scanner.nextInt();
        System.out.println(movingCount(m, n, k));
    }

    static int M = 0;
    static int N = 0;
    static int K = 0;
    static boolean[][] isVisited;

    public static int movingCount(int m, int n, int k) {
        M = m;
        N = n;
        K = k;
        isVisited = new boolean[m][n];
        return dfs(0, 0, 0, 0);
    }

    public static int dfs(int i, int j, int si, int sj) {
        if (i >= M || j >= N || si + sj > K || isVisited[i][j]) return 0;
        isVisited[i][j] = true;
        return 1 + dfs(i + 1, j, (i + 1) % 10 == 0 ? si - 8 : si + 1, sj) + dfs(i, j + 1, si, (j + 1) % 10 == 0 ? sj - 8 : sj + 1);
    }
}