BUPT OJ|311. 图像识别-计算机二2014

题目来源 北邮OJ

#include<bits/stdc++.h>
using namespace std;
/*Input:
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
*/
int T,N,M,D;
int data[105][105];
bool mark[105][105]; 
int route[8][2]={1,1,1,0,1,-1,0,-1,-1,-1,-1,0,-1,1,0,1};
// 选择C++11或14，否则会报编译错误
void DFS(int x,int y,int value){
	if(!(x>=0&&x<N&&y>=0&&y<M&&!mark[x][y]&&abs(value-data[x][y])<=D)) return;
	mark[x][y]=true;
	for(int i=0;i<8;i++){
		int xx=x+route[i][0];
		int yy=y+route[i][1];
		DFS(xx,yy,data[x][y]);
	}
}

int main()
{
	cin>>T;
	while(T--){
		cin>>N>>M>>D;
		memset(data,0,sizeof(data));
		memset(mark,false,sizeof(mark));
		for(int i=0;i<N;i++) {
			for(int j=0;j<M;j++){
				cin>>data[i][j];
			}
		}
		int cnt=0;
		for(int i=0;i<N;i++) {
			for(int j=0;j<M;j++){
				if(!mark[i][j]){
					DFS(i,j,data[i][j]);
					cnt++;
				}
			}
		}
		cout<<cnt<<endl;
	}
}