肿瘤诊断(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;
}