洛谷 P1162 填涂颜色

链接：https://www.luogu.org/problemnew/show/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饮品指出本题数据和数据格式不一样. 已修改(输入格式)

 

输出格式：

 

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

 

输入输出样例

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

题目中说了只有一个闭合圈，所以这个闭合圈中第一个需要变成2的0，他的上方是1，左方也是1，

所以找到这个点开始搜索能走到的点即是一个圈内的，然后把在这个圈内能走到的0都变成数字2，

然后输出即可；

#include<iostream>
#include<stdio.h>
#include<queue>
using namespace std;
int n;
int map[40][40];
int vis[40][40];
int dir[4][2] = { 1, 0, 0, -1, 0, 1, -1, 0 };
struct node{
	int x, y;
};
void bfs(int x, int y){
	queue<node>q;
	q.push({ x, y });
	while (!q.empty()){
		int x = q.front().x, y = q.front().y;
		q.pop();
		if (vis[x][y])continue;
		if (map[x][y] == 0)map[x][y] = 2; //在这个圈内能走到的0
		vis[x][y] = 1;
		for (int i = 0; i < 4; i++)
		{
			int xx = x + dir[i][0];
			int yy = y + dir[i][1];
			if (xx < n&&xx>0 && yy < n&&yy>0 && map[xx][yy] != 1)
				q.push({ xx, yy });
		}
	}
}
int main(){
	int flag=1;
	cin >> n;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			scanf("%d", &map[i][j]);
	for (int i = 0; i < n&&flag; i++)
		for (int j = 0; j < n;j++)
			if (map[i][j] == 0 && map[i][j - 1] == 1 && map[i - 1][j] == 1)
			{   //第一个满足条件的0，从该点开始搜索圈内
				bfs(i, j);
				flag = 0;
				break;
			}
	for (int i = 0; i < n; i++, cout << endl)
		for (int j = 0; j < n; j++)
			cout << map[i][j]<<" ";

}