Leetcode 827. 最大人工岛

给你一个大小为 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

思路：并查集快速维护集合。先将图中的1都插入p树中，然后遍历所有的0，搜寻其上下左右四个方向的联通块的大小，放入哈希表中（因为很可能多个方向的已经在一个联通块中了，哈希表可以避免重复。），随后判断一下包括自身（t = 1）所有的和，更新一下答案即可。注意 这里将二维坐标用get映射到了1维。

class Solution {
public:
    static const int N = 510 * 510;
    int p[N], s[N];
    int n, m;
    int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
    int find(int x) {
        if(p[x] != x) p[x] = find(p[x]);
        return p[x];
    }

    int get(int x, int y) {
        return x * m + y;
    }
    int largestIsland(vector<vector<int>>& grid) {
        n = grid.size(), m = grid[0].size();
        for(int i = 0; i < n * m; i ++ ) p[i] = i, s[i] = 1; // 注意枚举的范围，如果是
        int res = 1;
        for(int i = 0; i < n; i ++ )
            for(int j = 0; j < m; j ++ ) {
                if(grid[i][j]) {
                    int a = get(i, j);
                    for(int k = 0; k < 4; k ++ ) {
                        int x = i + dx[k], y = j + dy[k];
                        if(x >= 0 && x < n && y >= 0 && y < m && grid[x][y]) {
                            int b = get(x, y);
                            if(find(a) != find(b)) {  // 判断一下是不是已经在一起了
                                s[find(b)] += s[find(a)];
                                p[find(a)] = find(b);
                            }
                        }
                    }
                    res = max(res, s[find(a)]);
                }
            }

        for(int i = 0; i < n; i ++ )
            for(int j = 0; j < m; j ++ ) {
                if(!grid[i][j]) {
                    unordered_map<int, int> hash;
                    for(int k = 0; k < 4; k ++ ) {
                        int x = i + dx[k], y = j + dy[k];
                        if(x >= 0 && x < n && y >= 0 && y < m && grid[x][y]) {
                            int a = get(x, y);
                            hash[find(a)] = s[find(a)];
                        }
                    }
                    int t = 1;
                    for(auto [k, v] : hash) t += v;
                    res = max(res, t);
                }
            }
        return res;
    }
};