代码随想录 104. 建造最大岛屿

104. 建造最大岛屿

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

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

int lands(vector<vector<int>>& visit, vector<int> hash, int y, int x){
    int tmp = 1;
    if (y > 0){
        tmp += hash[visit[y - 1][x]];
        hash[visit[y - 1][x]] = 0;
    }
    if (x > 0){
        tmp += hash[visit[y][x - 1]];
        hash[visit[y][x - 1]] = 0;
    }
    if (y < visit.size() - 1){
        tmp += hash[visit[y + 1][x]];
        hash[visit[y + 1][x]] = 0;
    }
    if (x < visit[0].size() - 1){
        tmp += hash[visit[y][x + 1]];
        hash[visit[y][x + 1]] = 0;
    }
    return tmp;
}

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));
    vector<int> hash;
    hash.push_back(0);
    int mk = 1;
    for (int i = 0; i < n; i++){
        for (int j = 0; j < m; j++){
            cin >> mp[i][j];
        }
    }
    for (int i = 0; i < n; i++){
        for (int j = 0; j < m; j++){
            if (mp[i][j] == 1 && visit[i][j] == 0){
                int s = 0;
                dfs(mp, visit, i, j, s, mk);
                hash.push_back(s);
                mk++;
            }
        }
    }
    int mx = 0;
    if (hash[visit[0][0]] == n * m){
        mx = (n * m);
    }else{
        for (int i = 0; i < n; i++){
            for (int j = 0; j < m; j++){
                if (visit[i][j] == 0){
                    mx = max(mx, lands(visit, hash, i, j));
                }
            }
        }
    }
    cout << mx;
    return 0;
}

笔者自己的想法比较复杂且不好写，所以还是看了随想录的解法来写。随想录的解法是在visit中做标记时不以统一的0或1来标记是否访问，而是使用1，2，3……来对岛做进一步细分的标记，以标记作为索引建立哈希表，存储每个岛的面积，这样在主函数中第一遍遍历地图后，就能得到记录所有岛面积的哈希表，表中的索引与visit中陆地的标记相同，也就意味着在后续求取最大面积时只可以从visit中的陆地对应的标记直接访问到陆地所属岛屿的面积。在主函数中第一遍遍历后就来到了求取最大面积的部分。最大面积有两种情况，一种是整个地图都是陆地，就没有填补的空间，直接输出n*m即可，另外一种就是有海也有陆地，两种情况很好区分，随便取一个visit中的索引查询面积，若面积等于m*n就说明没有海。在有海的情况下，填一个陆地一定会大于不填的面积，所以只需要考虑海的部分即可。而面积的计算就会用到之前的哈希表，每个海的苏格方向都可能使陆地，所以只需要用所有相邻的visit的标记访问hash表拿到面积后相加，再+1即可，因为笔者的代码中，海的部分即使在dfs中访问过也依然标记为0，所以在建立hash表时先插入了一个0，使海在visit中的标记访问到的面积为0，这样在求取填一个海后的面积时可以直接求和。此外，因为一个海的相邻陆地中可能有2个及以上使属于同一个岛的，所以在每次求和后向hash表对应位置置0，一面重复技术，也是由于这个原因，求和部分单独写了一个函数，隔断主函数中的hash表与求和中的hash表，简化操作。

代码随想录 104. 建造最大岛屿