P1162 填涂颜色 -洛谷

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≤n≤30

解析

  要求在图里面找闭合图形,反向思维,找没有被1包围的0.
  可知边界必定没有被包围,就从四条边开始找到0就bfs或者dfs哗哗哗就可以了.

#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<queue>
#include<stdio.h>
#include<stdlib.h>
#include<map>
#ifndef NULL
#define NULL 0
#endif
using namespace std;

struct vec {
	int x,y;
}p{ -1,-1 };
int n,mp[31][31], dx[] = { 0,1,0,-1 }, dy[] = { 1,0,-1,0 };
void bfs()
{
	queue<vec >q;
	q.push(p);
	while (!q.empty()) {
		p = q.front(); q.pop();
		mp[p.x][p.y] = 3;
		for (int i = 0; i < 4; i++) {
			p.x += dx[i], p.y += dy[i];
			if (!mp[p.x][p.y] && p.x >= 0 && p.x < n&&p.y < n&&p.y >= 0)
				q.push(p);
			p.x -= dx[i], p.y -= dy[i];
		}
	}
}
int main()
{	

	cin >> n;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			cin >> mp[i][j];
	for (int i = 0; i < n; i++) {
		if (!mp[0][i]) {
			p.x = 0, p.y = i;
			bfs();
		}
		if (!mp[n - 1][i]) {
			p.x = n - 1, p.y = i;
			bfs();
		}
		if (!mp[i][0]) {
			p.x = i, p.y = 0;
			bfs();
		}
		else if (!mp[i][n - 1]) {
			p.x = i, p.y = n - 1;
			bfs();
		}
	}
	for(int i=0;i<n;i++){
		for (int j = 0; j < n; j++)
			if (mp[i][j] == 3)
				cout << 0 << ' ';
			else if (mp[i][j] == 0)
				cout << 2 << ' ';
			else
				cout << mp[i][j] << ' ';
		cout << endl;
	}
	return 0;
}