7-4 山 (100 分)java版

Drizzle 前往山地统计大山的数目，现在收到这片区域的地图，地图中用0（平地）和1（山峰）绘制而成，请你帮忙计算其中的大山数目山总是被平地四面包围着，每一座山只能在水平或垂直方向上连接相邻的山峰而形成。一座山峰四面被平地包围，这个山峰也算一个大山。另外，你可以假设地图的四面都被平地包围着。

要求:
输入：第一行输入M,N分别表示地图的行列，接下来M行每行输入N个数字表示地图
输出：输出一个整数表示大山的数目

示例:
输入：

4 5
1 1 0 0 0
1 1 0 0 0
0 0 1 0 0
0 0 0 1 1

输出：

3

范围:
对于 5% 的数据：M，N ≤ 10
对于 100% 的数据：M，N ≤ 2000

import java.io.*;

public class Main {

    public static void main(String[] args) throws IOException {
        StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        st.nextToken();
        int M = (int) st.nval;
        st.nextToken();
        int N = (int) st.nval;
        int nums[][] = new int[M][N];
        for (int i = 0; i < M; i++) {
            for (int j = 0; j < N; j++) {
                st.nextToken();
                nums[i][j] = (int) st.nval;
            }
        }
        if (nums.length == 0)
            System.out.println(0);
        else {
            int count = 0;
            for (int i = 0; i < M; i++) {
                for (int j = 0; j < N; j++) {
                    if (nums[i][j] == 1) {
                        count++;
                        dfs(nums, i, j);
                    }

                }
            }
            System.out.println(count);
        }
    }

    public static void dfs(int[][] nums, int i, int j) {
        int a = nums.length;
        int b = nums[0].length;
        if (i < 0 || j < 0 || i >= a || j >= b || nums[i][j] == 0) {
            return;
        }
        nums[i][j] = 0;
        dfs(nums, i - 1, j);
        dfs(nums, i + 1, j);
        dfs(nums, i, j - 1);
        dfs(nums, i, j + 1);
    }
}