题目
给定一个包含了一些 0 和 1的非空二维数组 grid
, 一个 岛屿 是由四个方向 (水平或垂直) 的 1
(代表土地) 构成的组合。你可以假设二维矩阵的四个边缘都被水包围着。
找到给定的二维数组中最大的岛屿面积。(如果没有岛屿,则返回面积为0。)
示例 1:
[[0,0,1,0,0,0,0,1,0,0,0,0,0], [0,0,0,0,0,0,0,1,1,1,0,0,0], [0,1,1,0,1,0,0,0,0,0,0,0,0], [0,1,0,0,1,1,0,0,1,0,1,0,0], [0,1,0,0,1,1,0,0,1,1,1,0,0], [0,0,0,0,0,0,0,0,0,0,1,0,0], [0,0,0,0,0,0,0,1,1,1,0,0,0], [0,0,0,0,0,0,0,1,1,0,0,0,0]]
对于上面这个给定矩阵应返回 6
。注意答案不应该是11,因为岛屿只能包含水平或垂直的四个方向的‘1’。
思路
并查集裸题
dfs搜索
代码
class Solution {
public:
class unionSet {
int ans;
vector<int> fa;
vector<int> size;
public:
unionSet(int n = 0):ans(0), fa(n), size(n, 1) {
for (int i = 0; i < n; i++) fa[i] = i;
}
int get(int x) {
return fa[x] = (fa[x] == x ? x : get(fa[x]));
}
void merge(int a, int b) {
int aa = get(a), bb = get(b);
if (aa == bb) return;
fa[aa] = bb;
size[bb] += size[aa];
}
int getAns() {
for (int i = 0; i < fa.size(); i ++) {
ans = max(ans, size[i]);
}
return ans;
}
};
#define ind(a, b, n) (((a) * (n)) + b)
int maxAreaOfIsland(vector<vector<int>>& grid) {
int n = grid.size(), m = grid[0].size(), flag = 0;
unionSet u(n * m);
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (grid[i][j] == 0) continue;
if (i + 1 < n && grid[i + 1][j] == 1) {
u.merge(ind(i, j, m), ind(i + 1, j, m));
}
if (j + 1 < m && grid[i][j + 1] == 1) {
u.merge(ind(i, j, m), ind(i, j + 1, m));
}
flag = 1;
}
}
return flag == 0 ? 0 : u.getAns();
}
};
执行用时 :16 ms, 在所有 C++ 提交中击败了95.45% 的用户
总结
有个小小的地方,就是处理全0的情况,因为我每个点默认size都是1,所以需要一个flag来标记都是0的情况
有点小小的难受。