建造最大岛屿-卡玛

题目链接: 建造最大岛屿

本题刚开始的想法是挨个将矩阵中为0的元素变为1,然后取计算最大岛屿的面积。这样做时间复杂度会达到O(n^4),代价有些过于高了。
代码随想录建造最大岛屿 给出的优化思路是只要用一次深搜把每个岛屿的面积记录下来就好。
第一步:一次遍历地图,得出各个岛屿的面积,并做编号记录。可以使用map记录,key为岛屿编号,value为岛屿面积
第二步:再遍历地图,遍历0的方格(因为要将0变成1),并统计该1(由0变成的1)周边岛屿面积,将其相邻面积相加在一起,遍历所有 0 之后,就可以得出 选一个0变成1 之后的最大面积。
在遍历中,实际上矩阵中每个节点都只遍历了一遍,总的时间复杂度为O(n^2)。

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

int dir[4][2] = {1, 0, 0, 1, -1, 0, 0, -1};
int num;

void dfs(vector<vector<int>>& graph, vector<vector<bool>>& visited, int x, int y, int mark){
    for(int i = 0; i < 4; i++){
        int nextx = x + dir[i][0];
        int nexty = y + dir[i][1];
        if(nextx < 0 || nextx >= graph.size() || nexty < 0 || nexty >= graph[0].size()){
            continue;
        }else{
            if(graph[nextx][nexty] == 1 && !visited[nextx][nexty]){
                visited[nextx][nexty] = true;
                num++;
                graph[nextx][nexty] = mark;
                dfs(graph, visited, nextx, nexty, mark);
            }
        }
    }
}

int main(){
    int n,m;
    cin >> n >> m;
    vector<vector<int>> graph(n, vector<int>(m));
    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++){
            cin >> graph[i][j];
        }
    }
    vector<vector<bool>> visited(n, vector<bool>(m, false));
    unordered_map<int, int> mp;
    int mark = 2;
    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++){
            if(graph[i][j] == 1 && !visited[i][j]){
                num = 0;
                visited[i][j] = true;
                num++;
                graph[i][j] = mark;
                dfs(graph, visited, i, j, mark);
                mp[mark] = num;
                mark++;
            }
        }
    }
    
    unordered_set<int> visitedSet;
    int ans = 0;
    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++){
            if(graph[i][j] == 0){
                visitedSet.clear();
                num = 1;
                for(int k = 0; k < 4; k++){
                    int nextx = i + dir[k][0];
                    int nexty = j + dir[k][1];
                    if(nextx < 0 || nextx >= graph.size() || nexty < 0 || nexty >= graph[0].size()){
                        continue;
                    }
                    if(visitedSet.find(graph[nextx][nexty]) != visitedSet.end()){
                        continue;
                    }
                    visitedSet.insert(graph[nextx][nexty]);
                    num += mp[graph[nextx][nexty]];
                }
            }
            ans = max(ans, num);
        }
    }
    cout << ans << endl;
    return 0;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值