简单的搜索——DFS(8)
题目来源:洛谷 P1596 [USACO10OCT]Lake Counting S
共十个测试点
题解
AC代码
#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
// 地图数组
char mp[maxn][maxn];
// 标记数组
int v[maxn][maxn];
int n, m;
// 方向数组
int dx[8] = {0, 1, 1, 1, 0, -1, -1, -1}, dy[8] = {1, 1, 0, -1, -1, -1, 0, 1};
int ans = 0;
struct point
{
int x, y;
};
// 初始化输入
void init()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
scanf("%s", mp[i] + 1);
}
// DFS 搜索
void dfs(int x, int y)
{
int tx, ty;
for (int i = 0; i < 8; ++i)
{
tx = x + dx[i], ty = y + dy[i];
if (mp[tx][ty] == '.' && v[tx][ty] == 0)
v[tx][ty] = 1;
// 如果 mp[x][y] == 'W' 将标记数组改为 1,并且 dfs(x,y)
else if (mp[tx][ty] == 'W' && v[tx][ty] == 0)
{
v[tx][ty] = 1;
dfs(tx, ty);
}
}
}
void solve()
{
// 遍历
for (int i = 1; i <= n; ++i)
{
for (int j = 1; j <= m; ++j)
{
// 如果 v[x][y] == 0 && mp[x][y] == 'W'
if (mp[i][j] == 'W' && v[i][j] == 0)
{
dfs(i, j);
++ans;
}
else if (mp[i][j] == '.')
v[i][j] = 1;
}
}
}
int main()
{
init();
solve();
printf("%d", ans);
return 0;
}