题目链接: 建造最大岛屿
本题刚开始的想法是挨个将矩阵中为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;
}