PAT甲级(1091)

PAT甲级(1091)_第1张图片

题意理解(个人看法)

1、每层MN个像素点共有L层,所以对一个MN*L的三维物体进行扫描,采用BFS进行搜索。
2、X、Y、Z数组中的1和-1代表前后左右上下方向,1表示前进一格想正方向搜索,-1表示后退一格向后方搜索。
3、先向X的正向搜索如果为1,cnt++,同时判断是否越界。

#include 

using namespace std;
struct node{
    int x,y,z;
};
int m,n,l,t;
int X[6] = {1,0,0,-1,0,0};
int Y[6] = {0,1,0,0,-1,0};
int Z[6] = {0,0,1,0,0,-1};

int arr[1300][130][80];
bool visit[1300][130][80];
bool judge(int x,int y,int z)
{
    if(x<0 || x>=m || y<0 || y>=n || z<0 || z>=l) return false;
    if(arr[x][y][z] == 0 || visit[x][y][z]==true) return false;
    return true;
}

int bfs(int x,int y,int z)
{
    int cnt=0;
    node tmp;
    tmp.x = x;tmp.y = y;tmp.z = z;
    queue<node> q;
    q.push(tmp);
    visit[x][y][z] = true;
    while(!q.empty())
    {
        node top = q.front();
        q.pop();
        cnt++;
        for(int i=0;i<6;i++)
        {
            int tx = top.x+X[i];
            int ty = top.y+Y[i];
            int tz = top.z+Z[i];
            if(judge(tx,ty,tz))
            {
                visit[tx][ty][tz] = true;
                tmp.x = tx;tmp.y = ty;tmp.z = tz;
                q.push(tmp);
            }
        }
    }
    if(cnt>=t) return cnt;
    else return 0;
}

int main()
{
    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",&arr[j][k][i]);
    int ans = 0;
    for(int i=0;i<l;i++)
        for(int j=0;j<m;j++)
            for(int k=0;k<n;k++){
                if(arr[j][k][i]==1 && visit[j][k][i]==false)
                {
                    ans+= bfs(j,k,i);
                }
            }
    cout << ans;
    return 0;
}

你可能感兴趣的:(PAT,甲级)