【华为OD题库-042】战场索敌-java

题目

有一个大小是N * M的战场地图，被墙壁’#‘分隔成大小不同的区域，上下左右四个方向相邻的空地∵，属于同一个区域，只有空地上可能存在敌人’E’，请求出地图上总共有多少区域里的敌人数小于K。
输入描述
第一行输入为 N M K;
N表示地图的行数，M表示地图的列数，K表示目标敌人数量N，M<=100
之后为一个N * M大小的字符数组
输出描述
敌人数小于K的区域数量
示例1:
输入:
3 5 2
…#EE
E.#E.
###…
输出
1
说明
地图被墙壁分为两个区域，左边区域有1个敌人，右边区域有3个敌人，符合条件的区域数量是1

思路

递归遍历解决

1. 如果某个位置不为‘#’,那么从此位置递归遍历，找上下左右不为‘#’的区域，计算其中的E的数量，并且将该区域标记为已访问
2. 下次遍历时，如果字符不为#且该位置未被遍历过，说明是一个新区域
3. 最后计算E数量小于K的区域个数

如果改变输入，可以将已访问过的区域直接标记为#
如果不改变输入，可以使用一个新的visited标记是否访问过该区域

题解

方案一：把已访问过的区域标记为#，改变了原有输入

package hwod;

import java.util.Arrays;
import java.util.Scanner;

public class BattlefieldFindEnemy {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int[] nums = Arrays.stream(sc.nextLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        int n = nums[0], m = nums[1], k = nums[2];
        char[][] arrs = new char[n][m];
        for (int i = 0; i < n; i++) {
            arrs[i] = sc.nextLine().toCharArray();
        }
        System.out.println(battleFieldFindEnemy(arrs, k));
    }
    
    private static int battleFieldFindEnemy(char[][] arrs, int k) {
        int ans = 0;
        for (int i = 0; i < arrs.length; i++) {
            for (int j = 0; j < arrs[0].length; j++) {
                if (arrs[i][j] != '#' && recur(arrs, i, j) < k) {
                    ans++;
                }
            }
        }
        return ans;
    }


    private static int recur(char[][] arrs, int i, int j) {
        int res = 0;
        if (i < 0 || j < 0 || i >= arrs.length || j >= arrs[0].length || arrs[i][j] == '#') return res;
        if (arrs[i][j] == 'E') res++;
        arrs[i][j] = '#';
        return res + recur(arrs, i + 1, j) + recur(arrs, i - 1, j) + recur(arrs, i, j + 1) + recur(arrs, i, j - 1);
    }
}

方案二：使用vistied数组标记某个位置是否被访问过，不改变原来的arrs

package hwod;

import java.util.Arrays;
import java.util.Scanner;

public class BattlefieldFindEnemy {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int[] nums = Arrays.stream(sc.nextLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        int n = nums[0], m = nums[1], k = nums[2];
        char[][] arrs = new char[n][m];
        for (int i = 0; i < n; i++) {
            arrs[i] = sc.nextLine().toCharArray();
        }
        System.out.println(battleFieldFindEnemy2(arrs, k));
    }

    private static int[][] visted;

    private static int battleFieldFindEnemy2(char[][] arrs, int k) {
        int ans = 0;
        visted = new int[arrs.length][arrs[0].length];
        for (int i = 0; i < arrs.length; i++) {
            for (int j = 0; j < arrs[0].length; j++) {
                if (visted[i][j] == 0 && arrs[i][j] != '#' && recur2(arrs, i, j) < k) {
                    ans++;
                }
            }
        }
        return ans;
    }

    private static int recur2(char[][] arrs, int i, int j) {
        int res = 0;
        if (i < 0 || j < 0 || i >= arrs.length || j >= arrs[0].length || visted[i][j] == 1 || arrs[i][j] == '#')
            return res;
        if (arrs[i][j] == 'E') res++;
        visted[i][j] = 1;
        return res + recur(arrs, i + 1, j) + recur(arrs, i - 1, j) + recur(arrs, i, j + 1) + recur(arrs, i, j - 1);
    }
}

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。