java P1162 填涂颜色

题目描述

由数字0组成的方阵中，有一任意形状闭合圈，闭合圈由数字1构成，围圈时只走上下左右4个方向。现要求把闭合圈内的所有空间都填写成2.例如：6×6的方阵（n=6），涂色前和涂色后的方阵如下：

0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 0 0 1
1 1 0 0 0 1
1 0 0 0 0 1
1 1 1 1 1 1
0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 2 2 1
1 1 1 1 1 1

输入格式

每组测试数据第一行一个整数n (1≤n≤30)

接下来n行，由0和1组成的n×n的方阵。

方阵内只有一个闭合圈，圈内至少有一个0。

//感谢黄小U饮品指出本题数据和数据格式不一样. 已修改(输入格式)

输出格式
已经填好数字22的完整方阵。

输入输出样例

输入 #1

6
0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 0 0 1
1 1 0 0 0 1
1 0 0 0 0 1
1 1 1 1 1 1

输出 #1

0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 2 2 1
1 1 1 1 1 1

输出 #1

0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 2 2 1
1 1 1 1 1 1

import java.util.*;

 
public class Main {
 
	static int[][] a = new int[32][32];
	static int[][] b = new int[32][32];
	
	static int[] dx = {0,-1,1,0,0};
	static int[] dy = {0,0,0,-1,1};
	static int n;
	
	static void dfs(int p, int q) {
		if (p < 0 ||p>n+1||q<0||q>n+1||a[p][q]!=0) {
			return;
		}
		a[p][q] = 1;
		for (int i = 1; i <= 4; i++) {
			dfs(p+dx[i], q+dy[i]);
		}
	}
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= n; j++) {
				b[i][j] = in.nextInt();
				if (b[i][j] == 0) {
					a[i][j] = 0;
				}
				else {
					a[i][j] = 2;
				}
			}
		}
		
		dfs(0, 0);
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= n; j++) {
				if (a[i][j] == 0) {
					System.out.print("2 ");
				}
				else {
					System.out.print(b[i][j]+" ");
				}
			}
			System.out.println();
		}
		
	}
	
 
}

思路： 找到方阵中的圈，圈外面的数圈外面的0数不变，里面的0 变为 1 ，整个方针外围外围放一圈0，才能找到正确的圈，也就是 1 ，