Flood fill算法——岛问题

//岛问题
//【题目】
//一个矩阵中只有0和1两种值，每个位置都可以和自己的上、下、左、右 四个位置相连，如
//果有一片1连在一起，这个部分叫做一个岛，求一个矩阵中有多少个岛?
//【举例】
//001010
//111010
//100100
//000000
//这个矩阵中有三个岛
//【进阶】
//如何设计一个并行算法解决这个问题
public class Islands {

	public static int countIslands(int[][] m) {
		if (m == null || m[0] == null) {
			return 0;
		}
		//N记录m的列数，M记录m的行数
		int N = m.length;
		int M = m[0].length;
		int res = 0;
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (m[i][j] == 1) {
					res++;
					infect(m, i, j, N, M);
				}
			}
		}
		return res;
	}

	//感染过程，将遍历过的1变为2
	//i用于遍历列，j用于遍历行
	public static void infect(int[][] m, int i, int j, int N, int M) {
		//如果i越界或j越界或某个位置不等于1，就直接返回
		if (i < 0 || i >= N || j < 0 || j >= M || m[i][j] != 1) {
			return;
		}
		m[i][j] = 2;
		//遍历列+1的位置
		infect(m, i + 1, j, N, M);
		//遍历列-1的位置
		infect(m, i - 1, j, N, M);
		//遍历行+1的位置
		infect(m, i, j + 1, N, M);
		//遍历行-1的位置
		infect(m, i, j - 1, N, M);
	}

	public static void main(String[] args) {
		int[][] m1 = {  { 0, 0, 0, 0, 0, 0, 0, 0, 0 }, 
				        { 0, 1, 1, 1, 0, 1, 1, 1, 0 }, 
				        { 0, 1, 1, 1, 0, 0, 0, 1, 0 },
				        { 0, 1, 1, 0, 0, 0, 0, 0, 0 }, 
				        { 0, 0, 0, 0, 0, 1, 1, 0, 0 }, 
				        { 0, 0, 0, 0, 1, 1, 1, 0, 0 },
				        { 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
		System.out.println(countIslands(m1));

		int[][] m2 = {  { 0, 0, 0, 0, 0, 0, 0, 0, 0 }, 
						{ 0, 1, 1, 1, 1, 1, 1, 1, 0 }, 
						{ 0, 1, 1, 1, 0, 0, 0, 1, 0 },
						{ 0, 1, 1, 0, 0, 0, 1, 1, 0 }, 
						{ 0, 0, 0, 0, 0, 1, 1, 0, 0 }, 
						{ 0, 0, 0, 0, 1, 1, 1, 0, 0 },
						{ 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
		System.out.println(countIslands(m2));

	}

}