肿瘤诊断(PAT)

题目链接：https://www.patest.cn/contests/gplt/L3-004

一道很裸的bfs，一开始以为会超时，抱着试一试的心态交了一发竟然过了

#include<iostream>
#include<algorithm>
#include<map>
#include<cstring>
#include<queue>
using namespace std;
int m,n,l,t;
int a[100][1300][130],vis[100][1300][130];
struct node{
	int i,j,k;
};
int ans;
queue<node> q;
int bfs(int k,int i,int j,int s){
	while(!q.empty())
	q.pop();
	vis[k][i][j]=1;
	node temp;
	temp.i=i;
	temp.j=j;
	temp.k=k;
	q.push(temp);
	s++;
	while(!q.empty()){
		node tt=q.front();
		q.pop();
		//cout<<tt.k<<" "<<tt.i<<" "<<tt.j<<endl;
		if(tt.k-1>=0&&a[tt.k-1][tt.i][tt.j]==1&&vis[tt.k-1][tt.i][tt.j]==0){
			s++;
			vis[tt.k-1][tt.i][tt.j]=1;
		node temp;
		temp.i=tt.i;
		temp.j=tt.j;
		temp.k=tt.k-1;
			q.push(temp);
		}
		if(tt.k+1<l&&a[tt.k+1][tt.i][tt.j]==1&&vis[tt.k+1][tt.i][tt.j]==0){
			s++;
			vis[tt.k+1][tt.i][tt.j]=1;
			node temp;
		temp.i=tt.i;
		temp.j=tt.j;
		temp.k=tt.k+1;
			q.push(temp);
		}
		
		if(tt.i-1>=0&&a[tt.k][tt.i-1][tt.j]==1&&vis[tt.k][tt.i-1][tt.j]==0){
			s++;
			vis[tt.k][tt.i-1][tt.j]=1;
			node temp;
		temp.i=tt.i-1;
		temp.j=tt.j;
		temp.k=tt.k;
			q.push(temp);
		}
		if(tt.i+1<m&&a[tt.k][tt.i+1][tt.j]==1&&vis[tt.k][tt.i+1][tt.j]==0){
			s++;
			vis[tt.k][tt.i+1][tt.j]=1;
			node temp;
		temp.i=tt.i+1;
		temp.j=tt.j;
		temp.k=tt.k;
			q.push(temp);
		}
		
		
		if(tt.j-1>=0&&a[tt.k][tt.i][tt.j-1]==1&&vis[tt.k][tt.i][tt.j-1]==0){
			s++;
			vis[tt.k][tt.i][tt.j-1]=1;
			node temp;
		temp.i=tt.i;
		temp.j=tt.j-1;
		temp.k=tt.k;
			q.push(temp);
		}
		if(tt.j+1<n&&a[tt.k][tt.i][tt.j+1]==1&&vis[tt.k][tt.i][tt.j+1]==0){
			s++;
			vis[tt.k][tt.i][tt.j+1]=1;
			node temp;
		temp.i=tt.i;
		temp.j=tt.j+1;
		temp.k=tt.k;
			q.push(temp);
		}
	}
	//cout<<s<<endl;
	if(s<t)
	return 0;
	
	return s;
}


int main(){
	cin>>m>>n>>l>>t;
	for(int k=0;k<l;k++){
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
				cin>>a[k][i][j];
			}
		}
	}
	for(int k=0;k<l;k++){
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
				if(a[k][i][j]==1&&vis[k][i][j]==0){
					ans+=bfs(k,i,j,0);
				}
			}
		}
	}
	cout<<ans<<endl;
  
  return 0;
}