PAT L3-004 腫瘤診斷(三維廣搜)

在診斷腫瘤疾病時，計算腫瘤體積是很重要的一環。給定病竈掃描切片中標註出的疑似腫瘤區域，請你計算腫瘤的體積。

輸入格式：

輸入第一行給出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

題意

如上

題解

三維廣搜，6個方向

代碼

 1 #include<stdio.h>
 2 #include<queue>
 3 #include<algorithm>
 4 using namespace std;  5 
 6 int a[61][130][1300];  7 int area=0;  8 int dx[6][3]={{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};//6個方向
 9 int n,m,l,t; 10 
11 struct p{ 12     int x,y,z; 13 }k,o; 14 
15 int check(int x,int y,int z){//x,y,z是否合法
16     if(x<0||y<0||z<0)return 0; 17     if(x>=l||y>=n||z>=m)return 0; 18     return 1; 19 } 20 
21 void bfs(int x,int y,int z){ 22     queue<p> qu; 23     int sum=1; 24     
25     k.x=x;k.y=y;k.z=z; 26     a[k.x][k.y][k.z]=0;//開始的點別忘了歸0 
27  qu.push(k); 28     
29     while(!qu.empty()){ 30         k=qu.front(); 31  qu.pop(); 32         for(int i=0;i<6;i++){ 33             o.x=k.x+dx[i][0]; 34             o.y=k.y+dx[i][1]; 35             o.z=k.z+dx[i][2]; 36             if(check(o.x,o.y,o.z)&&a[o.x][o.y][o.z]){ 37                 a[o.x][o.y][o.z]=0;//訪問過了就歸0
38                 sum++; 39  qu.push(o); 40  } 41  } 42  } 43     if(sum>=t) 44         area+=sum; 45 } 46 int main(){ 47     scanf("%d%d%d%d",&n,&m,&l,&t); 48     for(int k=0;k<l;k++){ 49         for(int i=0;i<n;i++){ 50             for(int j=0;j<m;j++) 51             scanf("%d",&a[k][i][j]); 52  } 53  } 54     
55     for(int k=0;k<l;k++){ 56         for(int i=0;i<n;i++){ 57             for(int j=0;j<m;j++) 58                 if(a[k][i][j]) 59  bfs(k,i,j); 60  } 61  } 62     printf("%d\n",area); 63     return 0; 64 }