NYOJ-92 图像有用区域（bfs）

图像有用区域

时间限制：3000 ms  |  内存限制：65535 KB
难度：4

描述

“ACKing”同学以前做一个图像处理的项目时，遇到了一个问题，他需要摘取出图片中某个黑色线圏成的区域以内的图片，现在请你来帮助他完成第一步，把黑色线圏外的区域全部变为黑色。


     


                图1                                                        图2 

已知黑线各处不会出现交叉（如图2）,并且，除了黑线上的点外，图像中没有纯黑色（即像素为0的点）。

输入第一行输入测试数据的组数N(0<N<=6)

每组测试数据的第一行是两个个整数W,H分表表示图片的宽度和高度(3<=W<=1440,3<=H<=960)

随后的H行，每行有W个正整数,表示该点的像素值。(像素值都在0到255之间，0表示黑色，255表示白色）

输出以矩阵形式输出把黑色框之外的区域变黑之后的图像中各点的像素值。
样例输入

1
5 5
100 253 214 146 120
123 0 0 0 0
54 0 33 47 0
255 0 0 78 0
14 11 0 0 0

样例输出

0 0 0 0 0
0 0 0 0 0
0 0 33 47 0
0 0 0 78 0
0 0 0 0 0

思路：考虑bfs扩展，不过本题容易忽视0的分布可能将图片分成几部分，导致bfs扩展时，无法扩展至另一部分，为了使0外部的区域连成一个整体，可以在图片的边缘加上一圈非0像素值。

AC代码：

#include <stdio.h>
#include <queue>

using namespace std;

struct point
{
int x;
int y;
};

int w, h;
int map[970][1450];
int dir[4][2] = {-1, 0, 0, 1, 1, 0, 0, -1};

void bfs(int a, int b)
{
queue <point> q;
point t1, t2;
t1.x = a;
t1.y = b;
q.push(t1);
while(!q.empty())
{
t1= q.front();
q.pop();
for(int i = 0; i < 4; ++i)
{
t2.x = t1.x + dir[i][0];
t2.y = t1.y + dir[i][1];
if(t2.x < 0 || t2.x > h+1 || t2.y < 0 || t2.y > w+1 || map[t2.x][t2.y] == 0)
continue;
map[t2.x][t2.y] = 0;
q.push(t2);
}
}
}

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&w,&h);
//边缘一圈赋值1，保证不为0
for(int i = 0; i <= h; ++i)
map[i][0] = map[i][w + 1] = 1;
for(int j = 0; j <= w; ++j)
map[0][j] = map[h + 1][j] = 1;

for(int i = 1; i <= h; ++i)
for(int j = 1; j <= w; ++j)
scanf("%d",&map[i][j]);
bfs(0, 0);
for(int i = 1; i <= h; ++i)
for(int j = 1; j <= w; ++j)
printf("%d%c",map[i][j],(j==w)?'\n':' ');
}
return 0;
}