L3-004 肿瘤诊断 (30分)
在诊断肿瘤疾病时,计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域,请你计算肿瘤的体积。
输入格式:
输入第一行给出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<iostream>
#include<vector>
using namespace std;
vector<vector<vector<int>>> photo;
vector<vector<int>> dir = { {1,0,0},{0,1,0},{0,0,1},{-1,0,0},{0,-1,0},{0,0,-1} };
int M, N, L, T;
int getArea(int k, int x, int y) {
if (k<0||k>=L|| //边界检测
x<0||x>=M||
y<0||y>=N||
photo[k][x][y] == 0 )//肿瘤查找
return 0;
photo[k][x][y] = 0; //访问过就消除
int result = 1; //累加统计
for (int i = 0; i < dir.size(); i++)
result += getArea(k + dir[i][0], x + dir[i][1], y + dir[i][2]);
return result;
}
int main() {
cin >> M >> N >> L >> T;
photo.resize(L);
for (int k = 0; k < L; k++) {
photo[k].resize(M);
for (int i = 0; i < M; i++) {
photo[k][i].resize(N);
for (int j = 0; j < N; j++) {
cin >> photo[k][i][j];
}
}
}
int result = 0;
for (int k = 0; k < L; k++) {
for (int i = 0; i < M; i++) {
for (int j = 0; j < N; j++) {
int area = getArea(k, i, j);
if (area >= T)
result += area;
}
}
}
cout << result;
return 0;
}