HDU 1241 dfs入门练手题

原创于 2022-08-04 20:51:32 发布 · 261 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先

ACM steps 专栏收录该内容

27 篇文章

订阅专栏

这段代码是用于解决一个名为'OilDeposits'的问题，它来源于HDOJ（杭州电子科技大学在线评测系统）。程序通过读取网格地图，使用深度优先搜索（DFS）算法来标记从油井（标记为'@'）扩散的所有位置，并计算出油藏的数量。内存限制为65MB，时间限制为2000ms。程序首先定义了方向数组dir，然后实现了一个ok函数来检查坐标是否在地图范围内。主函数中，它读取地图大小并遍历每个位置，遇到油井时调用dfs进行搜索，最后输出油藏的总数。

// Problem: Oil Deposits
// Contest: HDOJ
// URL: http://acm.hdu.edu.cn/showproblem.php?pid=1241
// Memory Limit: 65 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>

using namespace std;

const int N = 105;
int n, m;
char a[N][N];
int dir[8][2] = {{1, 0}, {-1, 0}, {1, 1}, {1, -1}, {-1, -1}, {-1, 1}, {0, 1}, {0, -1}};
bool ok(int x, int y)
{
    if (x < 1 || x > n || y < 1 || y > m)
        return false;
    return true;
}
void dfs(int x, int y)
{
    a[x][y] = '*';
    for (int i = 0; i < 8; i++)
    {
        x += dir[i][0];
        y += dir[i][1];
        if (ok(x, y) && a[x][y] == '@')
        {
            dfs(x, y);
        }
    }
}

int main()
{
    while (cin >> n >> m, n)
    {
        for (int i = 1; i <= n; i++)
        {
            for (int j = 1; j <= m; j++)
            {
                cin >> a[i][j];
            }
            getchar();
        }
        int count = 0;
        for (int i = 1; i <= n; i++)
        {
            for (int j = 1; j <= m; j++)
            {
                if (a[i][j] == '@')
                {
                    dfs(i, j);
                    count++;
                }
            }
        }
        printf("%d\n", count);
    }
}