二值矩阵求连通域

说白了就是BFS或者DFS，Leetcode刷题时做过了，不要怕。

• 这里增加了一个Point数据结构。
• 代码实现的四连通域，八连通域其实一样。
• 如果非要要求输入是图像，静下心，就是将二维数组替换了就行，Point结构要保留。

#include 
#include 
#include 

using namespace  std;

struct Point
{
    int r;
    int c;
    Point(int r_, int c_) : r(r_), c(c_) {}
    Point(const Point& p) : r(p.r), c(p.c) {}
};

class Solution
{
public:
    int m;
    int n;
    //判断索引点不要越界，该点之前没有访问过，该点是有效的点
    bool isvalid(int i, int j, vector>& matrix, vector>& mask) {
        return i >= 0 && i < m && j >= 0 && j < n && !mask[i][j] && matrix[i][j] == 1;
    }
    //将合适的点加入到队列中，并标记其被访问过
    void add(int i, int j, vector>& matrix, queue& q, vector>& mask) {
        if (isvalid(i, j, matrix, mask)) {
            q.push(Point(i, j));
            mask[i][j] = true;
        }
    }
    //主代码，两重for循环+一个queue
    vector> bwlabel(vector> &matrix) {
        m = matrix.size();
        n = matrix[0].size();
        vector> res;
        vector tmp;
        vector> mask(m, vector(n, false));
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (mask[i][j] || matrix[i][j] == 0)
                    continue;
                tmp.clear();
                queue q;
                q.push(Point(i, j));
                mask[i][j] = true;
                while (!q.empty()) {
                    Point t = q.front();
                    q.pop();
                    tmp.push_back(t);
                    //根据四邻域定义得来
                    add(t.r - 1, t.c, matrix, q, mask);
                    add(t.r + 1, t.c, matrix, q, mask);
                    add(t.r, t.c - 1, matrix, q, mask);
                    add(t.r, t.c + 1, matrix, q, mask);
                }
                res.push_back(tmp);
            }
        }
        return res;
    }
};

int main() {
    vector> m = {
        { 1,1,0,0,0 },
        { 1,0,1,0,0 },
        { 0,1,1,0,0 },
        { 0,0,0,1,0 },
        { 0,0,0,0,1 },
        { 0,0,0,0,0 }
    };
    vector> n = { {} };
    Solution s;
    vector> res = s.bwlabel(m);
    vector> rss = s.bwlabel(n);
    return 0;
}