【leetcode】695. 岛屿的最大面积

题目

给定一个包含了一些 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的情况
有点小小的难受。

上一篇:oracle-简单索引


下一篇:[CF13C] Sequence - dp