如果你一点也不了解什么是DFS(深度优先搜索),建议看一下这一篇LeetCode岛屿问题DFS
给你一个大小为 n x n 二进制矩阵 grid 。最多 只能将一格 0 变成 1 。
返回执行此操作后,grid 中最大的岛屿面积是多少?
岛屿 由一组上、下、左、右四个方向相连的 1 形成。
来源:力扣(LeetCode)最大人工岛
我们以下图来分析:
思路:我们不难想到,这一题主要就是改变一个海洋’0’变成陆地’1’,我们可以计算每个陆地的大小,然后在其周围寻找是否有相邻的陆地,相加再判断其大小是否是最大的,这样一个一个暴力遍历,就可以AC了。
我们首先要写一个搜寻该陆地的大小:
int max_size(vector<vector<int>>& grid,int i,int j,int n)
{
if(i<0 || i>=grid.size() || j<0 || j>=grid.size() || grid[i][j]!=1)
{
return 0;
}
grid[i][j]=n;
return 1 + max_size(grid,i+1,j,n)+
max_size(grid,i-1,j,n)+
max_size(grid,i,j+1,n)+
max_size(grid,i,j-1,n);
}
这里我们将已经遍历过的陆地更改为数字也是有讲究的:
假如一开始是这样的:
如果我们直接以它们的最大面积的大小来代替它们的陆地编号比如:
但是下面情况我们不好去处理:
所以,我们可以将更改的陆地数字作为下标,面积大小存储与数组中,与下标对应,这样我们就能很好处理上述,一块陆地计算两次情况。
接下来我们就
我的建议是遍历海洋格子,因为要继续遍历陆地格子,那么我们又要进DFS递归,这种递归的边界判断很难把握。
int judge(vector<vector<int>>& grid,vector<int>& arr,int i,int j)
{
int sum=0;
vector<int> _arr;
if(tof(grid,i-1,j) && find(_arr.begin(),_arr.end(),grid[i-1][j])==_arr.end() && grid[i-1][j]!=0)
{
sum+=arr[grid[i-1][j]-2];
_arr.push_back(grid[i-1][j]);
}
if(tof(grid,i+1,j) && find(_arr.begin(),_arr.end(),grid[i+1][j])==_arr.end() && grid[i+1][j]!=0)
{
sum+=arr[grid[i+1][j]-2];
_arr.push_back(grid[i+1][j]);
}
if(tof(grid,i,j-1) && find(_arr.begin(),_arr.end(),grid[i][j-1])==_arr.end() && grid[i][j-1]!=0)
{
sum+=arr[grid[i][j-1]-2];
_arr.push_back(grid[i][j-1]);
}
if(tof(grid,i,j+1) && find(_arr.begin(),_arr.end(),grid[i][j+1])==_arr.end() && grid[i][j+1]!=0)
{
sum+=arr[grid[i][j+1]-2];
_arr.push_back(grid[i][j+1]);
}
return sum;
}
我的思路就是直接暴力搜索上下左右,然后将已加过的陆地存入临时数组来判断,避免增加重复陆地的情况。
最终代码:
class Solution {
public:
//DFS返回面积
int max_size(vector<vector<int>>& grid,int i,int j,int n)
{
if(i<0 || i>=grid.size() || j<0 || j>=grid.size() || grid[i][j]!=1)
{
return 0;
}
grid[i][j]=n;
return 1 + max_size(grid,i+1,j,n)+
max_size(grid,i-1,j,n)+
max_size(grid,i,j+1,n)+
max_size(grid,i,j-1,n);
}
//判断海洋四周面积总和
int judge(vector<vector<int>>& grid,vector<int>& arr,int i,int j)
{
int sum=0;
vector<int> _arr;
if(tof(grid,i-1,j) && find(_arr.begin(),_arr.end(),grid[i-1][j])==_arr.end() && grid[i-1][j]!=0)
{
sum+=arr[grid[i-1][j]-2];
_arr.push_back(grid[i-1][j]);
}
if(tof(grid,i+1,j) && find(_arr.begin(),_arr.end(),grid[i+1][j])==_arr.end() && grid[i+1][j]!=0)
{
sum+=arr[grid[i+1][j]-2];
_arr.push_back(grid[i+1][j]);
}
if(tof(grid,i,j-1) && find(_arr.begin(),_arr.end(),grid[i][j-1])==_arr.end() && grid[i][j-1]!=0)
{
sum+=arr[grid[i][j-1]-2];
_arr.push_back(grid[i][j-1]);
}
if(tof(grid,i,j+1) && find(_arr.begin(),_arr.end(),grid[i][j+1])==_arr.end() && grid[i][j+1]!=0)
{
sum+=arr[grid[i][j+1]-2];
_arr.push_back(grid[i][j+1]);
}
return sum;
}
bool tof(vector<vector<int>>& grid,int i,int j)
{
if(i<0 || i>=grid.size() || j<0 || j>=grid.size())
{
return false;
}
return true;
}
int largestIsland(vector<vector<int>>& grid) {
vector<int> arr;
int x=2;
//第一次循环是用来找寻各个陆地面积,以及将面积大小存入以其陆地数字为下标的数组
for(int i=0;i<grid.size();i++)
{
for(int j=0;j<grid.size();j++)
{
if(grid[i][j]==1)
{
int sum=max_size(grid,i,j,x);
x++;
arr.push_back(sum);
}
}
}
//第二次循环是用来找寻最大面积
int max=0;
for(int i=0;i<grid.size();i++)
{
for(int j=0;j<grid.size();j++)
{
if(grid[i][j]==0)
{
int sum = judge(grid,arr,i,j);
if(sum>max)
{
max=sum;
}
}
}
}
if(arr.empty())
{
return 1;
}
if(max==0)
{
return grid.size()*grid.size();
}
return max<(grid.size()*grid.size())?max+1:max;
}
};
最后的判断,如果grid没有陆地,我们自己会填入一个陆地,所以要返回1。
max==0 是有可能没有海洋格子,循环没有进去,所以max=0,实则是满陆地格子
max<(grid.size()*grid.size())?max+1:max 这个要判断是因为我写的时候没有加入填的陆地,所以要判断一下。
我的写法只是我目前能想到的思路的直接写法,没有什么优化,期望大家有更优化的思路或者写法可以私信或评论区留言