PTA 肿瘤诊断
在诊断肿瘤疾病时,计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域,请你计算肿瘤的体积。
输入格式:
输入第一行给出4个正整数:M、N、L、T,其中M和N是每张切片的尺寸(即每张切片是一个M×N的像素矩阵。最大分辨率是1286×128);L(≤60)是切片的张数;T是一个整数阈值(若疑似肿瘤的连通体体积小于T,则该小块忽略不计)。
最后给出L张切片。每张用一个由0和1组成的M×N的矩阵表示,其中1表示疑似肿瘤的像素,0表示正常像素。由于切片厚度可以认为是一个常数,于是我们只要数连通体中1的个数就可以得到体积了。麻烦的是,可能存在多个肿瘤,这时我们只统计那些体积不小于T的。两个像素被认为是“连通的”,如果它们有一个共同的切面,如下图所示,所有6个红色的像素都与蓝色的像素连通。
输出格式:
在一行中输出肿瘤的总体积。
输入样例:
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
输出样例:
26
思路:
三维搜索就完事了,但是我不知道为啥**DFS**过不了!!!!!
代码如下:
#include<bits/stdc++.h>
using namespace std;
bool a[70][1300][200];
int M,N,L,T;
int p=1;
int dx[]= {1,-1,0,0,0,0};
int dy[]= {0,0,1,-1,0,0};
int dz[]= {0,0,0,0,1,-1};
void DFS(int x,int y,int z)
{
a[x][y][z]=0;
for(int i=0;i<6;i++)
{
int nx=x+dx[i],ny=y+dy[i],nz=z+dz[i];
if(0<=nx&&nx<L&&0<=ny&&ny<M&&0<=nz&&nz<N&&a[nx][ny][nz]==1)
{
DFS(nx,ny,nz);
p++;
}
}
return;
}
int main()
{
memset(a,0,sizeof(a));
int sum=0;
scanf("%d %d %d %d",&M,&N,&L,&T);
getchar();
for(int i=0;i<L;i++)
for(int j=0;j<M;j++)
for(int k=0;k<N;k++)
cin>>a[i][j][k];
for(int i=0;i<L;i++)
{
for(int j=0;j<M;j++)
{
for(int k=0;k<N;k++)
{
if(a[i][j][k]==1)
{
DFS(i,j,k);
if(p>=T)sum+=p;
p=1;
}
}
}
}
printf("%d",sum);
}
就这样了…
下面是其他博主用 BFS通过的代码:
#include<bits/stdc++.h>
using namespace std;
struct Node{
int x, y, z;
Node(int a, int b, int c)
{
x = a, y = b, z = c;
}
Node(){}
};
int g[1286][128][60] = {0}, sum = 0;
bool judge[1286][128][60] = {false};
int addx[6] = {1, -1, 0, 0, 0, 0};
int addy[6] = {0, 0, 1, -1, 0, 0};
int addz[6] = {0, 0, 0, 0, 1, -1};
int BFS(int a, int b, int c)
{
int num = 0;
Node s = Node(a, b, c);
queue<Node> q;
q.push(s);
judge[a][b][c] = true;
while(!q.empty())
{
Node front = q.front();
q.pop();
num++;
for(int i = 0; i < 6; i++)
{
s.x = front.x + addx[i];
s.y = front.y + addy[i];
s.z = front.z + addz[i];
if(s.x >= 0 && s.x < 1286 && s.y >= 0 && s.y < 128 && s.z >= 0 && s.z < 60)
{
if(!judge[s.x][s.y][s.z] && g[s.x][s.y][s.z])
{
q.push(s);
judge[s.x][s.y][s.z] = true;
}
}
}
}
return num;
}
int main(){
int m, n, l, t;
scanf("%d %d %d %d", &m, &n, &l, &t);
for(int i = 0; i < l; i++)
for(int j = 0; j < m; j++)
for(int k = 0; k < n; k++)
scanf("%d", &g[j][k][i]);
for(int i = 0; i < l; i++)
for(int j = 0; j < m; j++)
for(int k = 0; k < n; k++)
if(!judge[j][k][i] && g[j][k][i])
{
int ans = BFS(j, k, i);
if(ans >= t) sum += ans;
}
printf("%d",sum);
}