洛谷 P1162 填涂颜色（Java）

Posted 2022-05-24 爱敲代码的三毛

文章目录

• 填涂颜色
• • 题目描述
  • 输入格式
  • 输出格式
  • 样例 #1
  • • 样例输入 #1
    • 样例输出 #1
  • 思路
  • 代码

---

填涂颜色

题目描述

由数字$0$组成的方阵中，有一任意形状闭合圈，闭合圈由数字$1$构成，围圈时只走上下左右$4$个方向。现要求把闭合圈内的所有空间都填写成$2$.例如：$6\times 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\le n\le 30)$

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

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

输出格式

已经填好数字$2$的完整方阵。

样例 #1

样例输入 #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

思路

• dfs把1外围的所有0标记,因为1是只能上下左右的连接，所以不存在对角线
• 把没有标记的0填为2就好了

代码

import java.util.*;

public class Main 
    static int[][] upDownLeftRight = -1,0,1,0,0,-1,0,1;
    static int row;
    static int col;
    public static void dfs(boolean[][] flag, int[][] arr, int x, int y) 
        for (int i = 0; i < 4; i++) 
            int newX = x + upDownLeftRight[i][0];
            int newY =  y + upDownLeftRight[i][1];
            if (newX < 0 || newX >= row || newY < 0 || newY >= col || arr[newX][newY] == 1) 
                continue;
            
            if (!flag[newX][newY]) 
                flag[newX][newY] = true;
                dfs(flag, arr, newX, newY);
            

        
    
    public static void main(String[] args) 
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        row = n;
        col = n;
        int[][] arr = new int[n][n];
        boolean[][] flag = new boolean[n][n];

        for (int i = 0; i < n; i++) 
            for (int j = 0; j < n; j++) 
                arr[i][j] = sc.nextInt();
            
        
        // 第一行
        for (int i = 0; i < n; i++) 
            if (arr[0][i] == 0 && !flag[0][i]) 
                dfs(flag, arr, 0, i);
            
        
        // 最后一行
        for (int i = 0; i < n; i++) 
            if (arr[n-1][i] == 0 && !flag[n-1][i]) 
                dfs(flag, arr, n-1, i);
            
        
        // 第一列
        for (int i = 0; i < n; i++) 
            if (arr[i][0] == 0 && !flag[i][0]) 
                dfs(flag, arr, i, 0);
            
        
        // 最后一列
        for (int i = 0; i < n; i++) 
            if (arr[i][n-1] == 0 && !flag[i][n-1]) 
                dfs(flag, arr, i, n-1);
            
        
        for (int i = 0; i < n; i++) 
            for (int j = 0; j < n; j++) 
                if (arr[i][j] == 0 && !flag[i][j]) 
                    arr[i][j] = 2;
                
            
        
        for (int i = 0; i < n; i++) 
            for (int j = 0; j < n; j++) 
                if (j != n-1) 
                    System.out.print(arr[i][j]+" ");
                 else 
                    System.out.println(arr[i][j]);