北邮计算机2014机试 图像识别

题目:
Problem C. 图像识别
题目描述
在图像识别中,我们经常需要分析特定图像中的一些特征,而其中很重要的一点就是识别出图像的多个区域。在这个问题中,我们将给定一幅N x M的图像,其中每个1 x 1的点都用一个[0,255]的值来表示他的RGB颜色。如果两个相邻的像素点颜色差不超过D,我们就认为这两个像素点属于同一个区域。对于一个像素点(x,y) ,以下这8个点(如果存在)是与它相邻的:(x-1, y-1),(x-1,y),(x-1,y+1),(x,y-1),(x,y+1),(x+1,y-1),(x+1,y),(x+1,y+1)。
你的任务是写一个程序,分辨出给定图像中一共被分为多少个区域。
输入格式
输入数据包含多组测试数据。
输入的第一行是一个整数T(T<=100),表示测试数据的组数。
每组测试数据的第一行是三个整数N,M,D(1<=N,M<=100, 0<=D<=255),意义如上所述。
接下来N行,每行M个整数,表示给定图像的每个像素点颜色。
输出格式
对于每组测试数据输出一行,即图像中的区域数量。
输入样例
2
3 3 0
1 1 1
0 1 0
0 1 0
3 4 1
10 11 12 13
9 8 7 6
2 3 4 5
输出样例
3
1
题解:
本题实际是求连通块数量,遍历每个元素,如果这个元素未被访问,那这个元素一定是新的连通块的一部分,用DFS将连通块内的元素标记为访问。

#include<bits/stdc++.h>
using namespace std;
int  t,n,m,d,num;
int dx[8]={1,1,1,-1,-1,-1,0,0};
int dy[8]={0,-1,1,1,0,-1,1,-1};//方向打表 
int a[105][105];
bool b[105][105];//标记数组 
void dfs(int s,int e)
{//遍历该连通块(类似于染色) 
	b[s][e]=0;
	for(int i=0;i<8;i++)
	{
		int x=s+dx[i],y=e+dy[i];
		if(x>=1&&x<=n&&y>=1&&y<=m&&b[x][y]==1&&(abs(a[s][e]-a[x][y])<=d)&&a[x][y]!=-1)
		//一定要密切注意复杂条件的细节是否出错 
		    dfs(x,y);
	}
}
int main()
{
	cin>>t;
	while(t--)//Tips 
	{
		fill(a[0],a[0]+105*105,-1);
		fill(b[0],b[0]+105*105,1);//重要初始化 
		cin>>n>>m>>d;
		for(int i=1;i<=n;i++)
		    for(int j=1;j<=m;j++)
		       cin>>a[i][j];
	    num=0;
		for(int i=1;i<=n;i++)
		    for(int j=1;j<=m;j++)
		        if(b[i][j]==1)
				{
					num++;//连通块数量加1 
				    dfs(i,j);//遍历该连通块(染色) 
				}
		cout<<num<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38993096/article/details/88666947