P1162 填涂颜色（C++_BFS）

题目描述

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

反思

先完成后完美！从刚开始的120行+修改到现在的66行~~~

源码

#include
using namespace std;
int a[30][30], n, c[30][30] = { 0 };
int dir_x[4] = { 0,1,0,-1 };
int dir_y[4] = { 1,0,-1,0 };

class nodee
{
public:
	nodee() :x(-1), y(-1) {}
	int x, y;
};
queue<nodee> b;
void neww(int ii, int jj)
{
	nodee* t = new nodee;
	t->x = ii;
	t->y = jj;
	b.push(*t);
	c[t->x][t->y] = 1;
}
void first()
{
	for (int i = 0; i < n; i++)
	{
		if (!a[0][i] && c[0][i] == 0)//第0行
			neww(0, i);
		if (!a[i][0] && c[i][0] == 0)
			neww(i, 0);
		if (!a[n - 1][i] && c[n - 1][i] == 0)
			neww(n - 1, i);
		if (!a[i][n - 1] && c[i][n - 1] == 0)
			neww(i, n - 1);
	}
}
void bfs()
{
	while (!b.empty())
	{
		for (int i = 0; i < 4; i++)
			if (b.front().x + dir_x[i] >= 0 && b.front().x + dir_x[i] < n && b.front().y + dir_y[i] >= 0 && b.front().y + dir_y[i] < n && a[b.front().x + dir_x[i]][b.front().y + dir_y[i]] == 0 && c[b.front().x + dir_x[i]][b.front().y + dir_y[i]] != 1)
				neww(b.front().x + dir_x[i], b.front().y + dir_y[i]);
		b.pop();
	}
}
int main()
{
	cin >> n;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			cin >> a[i][j];
	first();//一圈
	bfs();
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			if (!c[i][j] && a[i][j] == 0)
				cout << 2 << " ";
			else
				cout << a[i][j] << " ";
		}
		cout << endl;
	}
	return 0;
}