洛谷題單指南-搜尋-P1596 [USACO10OCT] Lake Counting S

江城伍月發表於2024-03-07

原題連結:https://www.luogu.com.cn/problem/P1596

題意解讀:本題就是計算連通分量的數量,透過DFS洪水填充即可。

解題思路:

遍歷每一個格子,如果沒有標記過且有水,則對其進行標記,再列舉8個方向的格子,判斷是否同樣沒有水且沒有標記過,DFS遞迴處理。

100分程式碼:

#include <bits/stdc++.h>
using namespace std;

const int N = 105;

char a[N][N];
bool flag[N][N];

int dx[8] = {-1, -1, 0, 1, 1, 1, 0, -1};
int dy[8] = {0, -1, -1, -1, 0, 1, 1, 1};

int n, m;
int ans;

void dfs(int x, int y)
{
    flag[x][y] = true;
    for(int i = 0; i < 8; i++)
    {
        int nx = x + dx[i], ny = y + dy[i];
        if(nx >= 1 && nx <= n && ny >= 1 && ny <= m && a[nx][ny] == 'W' && !flag[nx][ny])
        {
            dfs(nx, ny);
        }
    }
}

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            cin >> a[i][j];
    
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {
            if(a[i][j] == 'W' && !flag[i][j])
            {
                dfs(i, j); //dfs染色
                ans++;
            }
        }
    }
    cout << ans;

    return 0;
}

相關文章