java实现—PTA打卡问题

7-2 666 (25 分)

小明有一张m*n的好习惯记录卡，记录每一天的好习惯目标达成度（数字0-9表示）。某天目标完成达成，就在当天的格子里写上数字6，目标没有完全达成就写上一个小于6的数字（0-5），目标超额完成就写上一个大于6的数字（7-9）。记录卡上如果能找到一条长度为3的路径并且路径上的三个数字都大于等于6（这里的路径是指从某个格子出发，可以向左、右、上、下格子移动，并且不能重复经过一个格子），则小明就能得到一个“666”奖励。
请你帮小明统计下他总共能得到多少“666”奖励。

输入格式:
输入第一行给出两个正整数m,n(1=<m,n<=100),随后是m行，每行包含n个0-9之间的数字。

输出格式:
先输出m行，每行包括n个整数，代表从当前格子出发得到的“666”奖励个数，中间用空格分割，最后一个数字后面不带空格。然后再在下一行输出得到的“666”奖励总数。

输入样例:

3 3
6 6 7
3 8 3
7 9 5

输出样例:

2 1 2
0 3 0
1 1 0
10

import java.util.Scanner;

public class Work666 {
    private static int[][] resource = new int[200][200];//输入的数组
    private static int[][] result = new int[200][200];//保存的结果
    private static int n;//行
    private static int m;//列

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        n = scan.nextInt();//输入行
        m = scan.nextInt();//输入列

        for (int i = 1; i <= n; i++) {//输入数据
            for (int j = 1; j <= m; j++) {
                resource[i][j] = scan.nextInt();
            }
        }

        /*只有大于等于6的才计算，如果小于6结果就标记0*/
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                if (resource[i][j] >= 6)
                    result[i][j] = backTrack(i, j, 1, 0);
                else
                    result[i][j] = 0;
            }
        }

        /*输出结果*/
        int sum = 0;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                System.out.printf("%d ", result[i][j]);
                sum += result[i][j];
            }
            System.out.println();
        }
        System.out.println(sum);

    }

    public static int backTrack(int i, int j, int number, int direction) {
        /*起始格子只有大于6才可以进入；
        i,j表示坐标，number表示经过的格子的个数，direction表示方向；
        1表示想右，2表示向下，3表示向左，4表示向上
         */
        int nu = 0, count = 0;//nu表示有符合的解,count用于记录总的解个数

        if (resource[i][j + 1] >= 6 && direction != 3) {//向右移动，且不往左移
            if (number == 2) {
                nu++;
            } else {
                count += backTrack(i, j + 1, number + 1, 1);//往右移，经过格子数+1
            }
        }

        if (resource[i + 1][j] >= 6 && direction != 4) {//向下移动，且不往上移
            if (number == 2) {//经过2个格子，到达第3个格子(算上自己)
                nu++;
            } else {
                count += backTrack(i + 1, j, number + 1, 2);//往下移，经过格子数+1
            }
        }

        if (resource[i][j - 1] >= 6 && direction != 1) {//向左移动，且不往右移
            if (number == 2) {//经过2个格子，到达第3个格子(算上自己)
                nu++;
            } else {
                count += backTrack(i, j - 1, number + 1, 3);//往左移，经过格子数+1
            }
        }

        if (resource[i - 1][j] >= 6 && direction != 2) {//向上移动，且不往下移
            if (number == 2) {//经过2个格子，到达第3个格子(算上自己)
                nu++;
            } else {
                count += backTrack(i - 1, j, number + 1, 4);//往上移，经过格子数+1
            }
        }

        if (number == 2) {
            return nu;
        } else {
            return count;
        }
    }
}