PAT (Advanced) 1091. Acute Stroke (30)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/cx86918626/article/details/79946109

原题:1091. Acute Stroke (30)



解题思路:

按题意进行bfs即可(dfs会爆栈),求出值为1的联通分量,数量大于等于阈值的加入最终答案。


代码如下:

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int maxm = 1300;
const int maxn = 130;
const int maxl = 65;
int G[maxn][maxm][maxl];
int vis[maxn][maxm][maxl];
int m, n, l, t;
int dx[] = {1, 0, 0, -1, 0, 0};
int dy[] = {0, 1, 0, 0, -1, 0};
int dz[] = {0, 0, 1, 0, 0, -1};

bool is_valid(int x, int y, int z)
{
    return x >= 0 && x < m && y >= 0 && y < n && z >= 0 && z < l;
}

struct Ax
{
    int x, y, z;
    Ax(int _x, int _y, int _z): x(_x), y(_y), z(_z) {}
};

void bfs(int i, int j, int k, int& cnt)
{
    queue<Ax> q;
    q.push(Ax(i, j, k));
    vis[i][j][k] = 1;
    cnt++;
    while(!q.empty())
    {
        Ax top = q.front(); q.pop();
        for(int it = 0; it < 6; it++)
        {
            int x = top.x + dx[it];
            int y = top.y + dy[it];
            int z = top.z + dz[it];
            if(is_valid(x, y, z) && !vis[x][y][z] && G[x][y][z])
            {
                cnt++;
                vis[x][y][z] = 1;
                q.push(Ax(x, y, z));
            }
        }
    }
}


int main()
{
    while(scanf("%d%d%d%d", &m, &n, &l, &t) == 4)
    {
        for(int k = 0; k < l; k++)
            for(int i = 0; i < m; i++)
                for(int j = 0; j < n; j++)
                    scanf("%d", &G[i][j][k]);
        memset(vis, 0, sizeof(vis));
        int ans = 0;
        for(int k = 0; k < l; k++)
            for(int i = 0; i < m; i++)
                for(int j = 0; j < n; j++)
                {
                    int cnt = 0;
                    if(!vis[i][j][k] && G[i][j][k])
                    {
                        bfs(i, j, k, cnt);
                        if(cnt >= t)
                            ans += cnt;
                    }
                }
        printf("%d\n", ans);
    }
    return 0;
}

阅读更多

没有更多推荐了,返回首页