【Leetcode】305. Number of Islands II

题目地址：

https://leetcode.com/problems/number-of-islands-ii/

给定一个二维零矩阵，再给点一个数组，数组每个位置是个长为$2$的整数数组，是矩阵里的一个坐标。遍历数组的同时，会把矩阵相应坐标位置上的$0$变为$1$。要求在线计算遍历数组的时候连通块（岛屿）的数量。

思路是并查集。在遍历数组的时候，取出坐标，如果矩阵在该位置已经有岛屿则略过（这里主要是预防数组里有重复坐标，也就是同一个位置变了两次$1$）；如果没有岛屿，则先将岛屿数加$1$，然后将这个岛屿和四周的岛屿全做一遍union，每union一次就要将岛屿数减去$1$，同时动态将岛屿数加入最后答案即可。代码如下：

class Solution {
 public:
  vector<int> p;
  int cnt;
  int find(int x) { return x == p[x] ? x : p[x] = find(p[x]); }
  void merge(int x, int y) {
    int px = find(x), py = find(y);
    if (px == py) return;
    p[px] = py;
    // 每次合并一次岛屿，总岛屿数量就少1
    cnt--;
  }

  vector<int> numIslands2(int m, int n, vector<vector<int>>& ps) {
    p.resize(m * n, -1);
    static int d[] = {-1, 0, 1, 0, -1};
    auto f = [&](int x, int y) { return x * n + y; };
    vector<int> res;
    for (auto& po : ps) {
      int x = po[0], y = po[1];
      int u = f(x, y);
      // 如果u的位置本身就是陆地了，则不需要再变一次
      if (!~p[u]) {
        // 先变成陆地
        p[u] = u;
        cnt++;
        for (int k = 0; k < 4; k++) {
          int nx = x + d[k], ny = y + d[k + 1];
          int v = f(nx, ny);
          if (0 <= nx && nx < m && 0 <= ny && ny < n && ~p[v]) merge(u, v);
        }
      }
      res.push_back(cnt);
    }

    return res;
  }
};

时间复杂度$O(k{\log }^{\ast }k)$（${\log }^{\ast }$是iterative logarithm，$k$是positions的长度），空间复杂度$O(mn)$。