给你一个大小为 n x n 二进制矩阵 grid 。最多 只能将一格 0 变成 1 。
返回执行此操作后,grid 中最大的岛屿面积是多少?
岛屿 由一组上、下、左、右四个方向相连的 1 形成。
示例 1:
输入: grid = [[1, 0], [0, 1]]
输出: 3
解释: 将一格0变成1,最终连通两个小岛得到面积为 3 的岛屿。
示例 2:
输入: grid = [[1, 1], [1, 0]]
输出: 4
解释: 将一格0变成1,岛屿的面积扩大为 4。
示例 3:
输入: grid = [[1, 1], [1, 1]]
输出: 4
解释: 没有0可以让我们变成1,面积依然为 4。
提示:
n == grid.length
n == grid[i].length
1 <= n <= 500
grid[i][j] 为 0 或 1
class Solution {
public:
vector<int> d = {0, -1, 0, 1, 0};
unordered_map<int, int> area;
int dfs(vector<vector<int>>& grid, vector<vector<int>> &flag, int t, int x, int y) {
int n = grid.size();
int num = 1;
flag[x][y] = t;
for (int i = 0; i < 4; i++) {
int x2 = x + d[i];
int y2 = y + d[i + 1];
if (x2 >= 0 && x2 < n && y2 >= 0 && y2 < n && flag[x2][y2] == 0 && grid[x2][y2] == 1) {
num += dfs(grid, flag, t, x2, y2);
}
}
return num;
}
int largestIsland(vector<vector<int>>& grid) {
int n = grid.size(), isize = 0;
vector<vector<int>> flag(n, vector<int>(n));
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 1 && flag[i][j] == 0) {
int t = n * i + j + 1;
area[t] = dfs(grid, flag, t, i, j);
isize = max(isize, area[t]);
}
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 0) {
int z = 1;
unordered_set<int> connected;
for (int k = 0; k < 4; k++) {
int x = i + d[k], y = j + d[k + 1];
if (!(x >= 0 && x < n && y >= 0 && y < n) || flag[x][y] == 0 || connected.count(flag[x][y]) > 0) {
continue;
}
z += area[flag[x][y]];
connected.insert(flag[x][y]);
}
isize = max(isize, z);
}
}
}
return isize;
}
};