洛谷-P1451-求细胞数量

题目描述

一矩形阵列由数字0到9组成，数字1到9代表细胞，细胞的定义为沿着细胞数字上下左右若还是细胞数字则为同一细胞，求给定矩形阵列的细胞个数

输入格式

第一行两个整数代表矩阵大小n和m。
接下来n行，每行一个长度为m的只含字符0到9的字符串，代表这个n x m的矩阵

输入格式

一行一个整数代表细胞个数

输入输出样例

输入#1

4 10
0234500067
1034560500
2045600671
0000000089

输出#1

4

Java代码

import java.util.Scanner;

public class P1451 {
    static int[][] direction = {{1,0}, {-1,0},{0,1},{0,-1}};
    static String[][] arr;
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int count = 0;
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        arr = new String[n][m];
        for (int i = 0; i < n; i++) {
            arr[i] = scanner.next().split("");
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if(!arr[i][j].equals("0")){
                    count++;
                    arr[i][j] = "0";
                    dfs(i, j);
                }
            }
        }
        System.out.println(count);
    }
    static void dfs(int x, int y){
        for (int i = 0;i < direction.length;i++){
            int dx = x +direction[i][0];
            int dy = y + direction[i][1];
            if(dx < 0 || dy < 0 || dx >= arr.length || dy >= arr[0].length || "0".equals(arr[dx][dy])) continue;
            arr[dx][dy] = "0";
            dfs(dx, dy);
        }
    }
}