PAT.A1091 Acute Stroke

返回目录

题意

给出一一个三维数组，数组元素的取值为0或1。与某一个元素相邻的元素为其上、下、左、右、前、后这6个方向的邻接元素。另外，若干个相邻的“1”称为一一个“块”(不必两两相邻，只要与块中某一个“1”相邻，该“1”就在块中)。而如果某个块中的“1”的个数不低于T个，那么称这个块为“卒中核心区”。现在需要求解所有卒中核心区中的1的个数之和。

样例（可复制）

3 4 5 2
1 1 1 1
1 1 1 1
1 1 1 1
0 0 1 1
0 0 1 1
0 0 1 1
1 0 1 1
0 1 0 0
0 0 0 0
1 0 1 1
0 0 0 0
0 0 0 0
0 0 0 1
0 0 0 1
1 0 0 0

//output
26

注意点

1. 本题思路如下：①枚举三维数组中的每一个位置，如果为0，则跳过;如果为1,则使用BFS查询与该位置相邻的6个位置(前提是不出界)，判断它们是否为1 (如果某个相邻的位置为1,则同样去查询与该位置相邻的6个位置，直到整个“1”块访问完毕)。②为了防止重复，设置了一个flag的3维bool数组，标记当前位置是否被访问过
2. 三维01矩阵不可设置为bool型，否则最后两组数据会答案错误。
3. 本题使用DFS非常容易在最后两组数据中出现段错误，原因是当三维矩阵中所有元素均为1时，DFS的深度过深，会使系统栈达到上限，从而爆栈。
4. 输入数据时是按多个二维矩阵的方式读入的，因此3层for循环中的第一层需要遍历矩阵编号，第二、三层才是单个矩阵的数据读入。

#include<bits/stdc++.h>
using namespace std;

struct Node{
	int x,y,z;
}node,top;
int n,m,s,t,ans=0;//n*m*s为三维矩阵，t为阈值下限 
int tensor[1300][130][61];
bool flag[1300][130][61]={false};//标记当前位置是否被访问过 
int xx[6]={0,0,0,0,1,-1},yy[6]={0,0,1,-1,0,0},zz[6]={1,-1,0,0,0,0};
bool judge(int x,int y,int z) {
	if(x>=n||x<0||y>=m||y<0||z>=s||z<0)return false;
	if(tensor[x][y][z]==0||flag[x][y][z])return false;
	return true;
}
int BFS(int x,int y,int z){
	int sum=0;//当前块中1的数量 
	queue<Node> q;
	node.x=x,node.y=y,node.z=z;
	q.push(node);
	flag[x][y][z]=true;
	while(!q.empty()){
		top=q.front();
		q.pop();
		sum++;
		for(int i=0;i<6;i++){
			int newx=top.x+xx[i];
			int newy=top.y+yy[i];
			int newz=top.z+zz[i];
			if(judge(newx,newy,newz)){
				node.x=newx,node.y=newy,node.z=newz;
				q.push(node);
				flag[newx][newy][newz]=true;
			}
		}
	}
	if(sum>=t)return sum;
	else return 0;
}
int main(){
	cin>>n>>m>>s>>t;
	for(int z=0;z<s;z++){
		for(int x=0;x<n;x++){
			for(int y=0;y<m;y++){
				scanf("%d",&tensor[x][y][z]);
			}
		}
	}
	for(int z=0;z<s;z++){
		for(int x=0;x<n;x++){
			for(int y=0;y<m;y++){
				if(tensor[x][y][z]==1&&!flag[x][y][z])ans+=BFS(x,y,z);
			}
		}
	}
	cout<<ans;
    return 0;
}