https://leetcode.cn/problems/making-a-large-island/ 给你一个大小为 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

# 题解

const vector<int> d = {0, -1, 0, 1, 0};

class Solution {
public:
bool valid(int n, int x, int y) {
return x >= 0 && x < n && y >= 0 && y < n;
}

int dfs(const vector<vector<int>> &grid, int x, int y, vector<vector<int>> &tag, int t) {
int n = grid.size(), res = 1;
tag[x][y] = t;
for (int i = 0; i < 4; i++) {
int x1 = x + d[i], y1 = y + d[i + 1];
if (valid(n, x1, y1) && grid[x1][y1] == 1 && tag[x1][y1] == 0) {
res += dfs(grid, x1, y1, tag, t);
}
}
return res;
}

int largestIsland(vector<vector<int>>& grid) {
int n = grid.size(), res = 0;
vector<vector<int>> tag(n, vector<int>(n));
unordered_map<int, int> area;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 1 && tag[i][j] == 0) {
int t = i * n + j + 1;
area[t] = dfs(grid, i, j, tag, t);
res = max(res, 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 (!valid(n, x, y) tag[x][y] == 0 connected.count(tag[x][y]) > 0) {
continue;
}
z += area[tag[x][y]];
connected.insert(tag[x][y]);
}
res = max(res, z);
}
}
}
return res;
}
};

复杂度分析:

  • 时间复杂度 \(O (n^2)\)
  • 空间复杂度 \(O (n^2)\)