代码随想录 102. 沉没孤岛

102. 沉没孤岛

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

void dfs(vector<vector<int>>& mp, vector<vector<int>>& visit, int y, int x){
    if (mp[y][x] == 0 || visit[y][x] == 1) return;
    if (mp[y][x] == 1 && visit[y][x] == 0) visit[y][x] = 1;
    
    if (y > 0) dfs(mp, visit, y - 1, x);
    if (x > 0) dfs(mp, visit, y, x - 1);
    if (y < mp.size() - 1) dfs(mp, visit, y + 1, x);
    if (x < mp[0].size() - 1) dfs(mp, visit, y, x + 1);
}

int main(){
    int n, m;
    cin >> n >> m;
    vector<vector<int>> mp(n, vector<int>(m, 0));
    vector<vector<int>> visit(n, vector<int>(m, 0));
    for (int i = 0; i < n; i++){
        for (int j = 0; j < m; j++){
            cin >> mp[i][j];
        }
    }
    for (int i = 0; i < m; i++){
        dfs(mp, visit, 0, i);
        dfs(mp, visit, n - 1, i);
    }
    for (int i = 1; i < n - 1; i++){
        dfs(mp, visit, i, 0);
        dfs(mp, visit, i, m - 1);
    }
    for (int  i = 0; i < n; i++){
        for (int  j = 0; j < m; j++){
            cout << visit[i][j] << " ";
        }
        cout << "\n";
    }
    
    return 0;
}

与之前的求孤岛面积类似，都是在与边缘相邻上做文章，不过看了随想录的思路后笔者发觉之前求面积时添加一个标记是否与边缘相邻的参数属于是脱裤子放屁，其实只要在住函数中把边缘的元素都放进遍历函数狗过一遍即可。说回这题，只需要在主函数中把四条边上的元素丢进dfs中一遍就行，一开始笔者想的是在输出时将mp数组与visit数组逐位相乘输出，visit中只对与边缘相邻的陆地做了标记，能把mp中的孤岛变为0，后来一想，既然笔者没有对mp中表示海的0对应的visit中的位置做置1处理，那么其实直接输出visit数组就行了。

代码随想录 102. 沉没孤岛