OJ笔记 18719 填涂颜色

18719 填涂颜色

Description
由数字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。

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

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

来源 洛谷 P1162

思路：

1. 创建一个辅助矩阵，输入时把原来矩阵0的位置染0，1的位置染2。
2. 然后递归搜索辅助矩阵染色把圈外的0全染成1；
3. 遍历打印时：把辅助矩阵0的位置染成2输出，其他位置按照原来的矩阵输出。

#include <iostream>
#include <cstdio>

using namespace std;

int a[100][100]; //输入的数组
int b[100][100]; //辅助数组
int n;//矩阵维数

void dfs(int p,int q)//p:上下遍历索引 q:左右遍历索引
{
    //递归出口:如果搜过头或者已经被搜过了或者本来就是墙的就往回（注意边界！！）
    if(p < 0 || p > n + 1 || q < 0 || q > n + 1 || b[p][q] != 0){
        return;
    }
    b[p][q] = 1;//把闭合圈外的0染成1
    //往四个方向继续搜索染色
    dfs(p - 1,q);//向上
    dfs(p + 1,q);//向下
    dfs(p,q - 1);//向左
    dfs(p,q + 1);//向右
}

int main()
{
    scanf("%d",&n);//输入方阵维数
    //输入初始颜色方阵
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++){
            cin >> a[i][j];
            if(a[i][j] == 0){
                b[i][j] = 0;//0的地方辅助方阵对应也为0
            }
            else{
                b[i][j] = 2;//把1的地方全涂成2
            }
        }
    }
    //从（0，0）开始搜索，把b闭合圈外的0涂成1
    dfs(0,0);

    //打印
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++){
            if(b[i][j] == 0){ //b中原来是0的没被染成1，就是在闭合圈内的，打印2
                cout << 2 << " ";
            }
            else{ //其他地方按原来输入的（数组a）输出
                cout << a[i][j] << " ";
            }
        }
        cout << endl; //一列结束，换行
    }

    return 0;
}