PAT (Advanced) 1091. Acute Stroke (30)

最新推荐文章于 2024-04-28 23:35:45 发布

原创最新推荐文章于 2024-04-28 23:35:45 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

PAT 专栏收录该内容

104 篇文章

订阅专栏

本文介绍了解决 AcuteStroke 问题的方法，通过使用广度优先搜索 (BFS) 来找出连通分量中值为1的元素数量，并统计大于等于阈值的分量总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题：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;
}