Sicily 相连的1 | 算法期末机考模拟题

Description

对于一个01矩阵A，求其中有多少片连成一片的1. 每个1可以和上下左右的1相连.

请为下面的Solution类实现解决这一问题的函数countConnectedOnes，函数参数A为给出的01矩阵，A的行数和列数均不大于1000. 函数的返回值是问题的答案.

class Solution {
public:
    int countConnectedOnes(vector<vector <char>>& A) {
    }
};

例1：
A=
100
010
001
答案为3.

例2：
A=
1101
0101
1110
答案为2.

Thinking

依题意需要查找一个矩阵中的连通分量，从坐标[0][0]的点开始对矩阵DFS遍历，遇到为1的点时，联通分量的个数count加一，将该点的值修改为3，并对该点前后左右的点继续进行DFS；遇到为0或为3的点则跳过。
一开始我只设定DFS的方向朝右下方进行，忽略了可能与左下方相连的点，导致联通分量数目增加，后来修改为对(k + 1, j)，(k - 1, j)，(k, j+ 1)，(k, j - 1)四个方向的相邻点都遍历。

Solution

class Solution {
public:
    int countConnectedOnes(vector<vector<char> >& A) {
        if(A.size() == 0) return 0;
        int count = 0;
        for(int i = 0; i < A.size(); i++){
            for(int j = 0; j < A[0].size(); j++){
                if(A[i][j] == '1' ){
                    count++;
                    connect(A, i, j);   
                }
            }
        }
        return count;
    }
    void connect(vector<vector<char> > & A, int k, int l){
        A[k][l] = '3';
        if(k < A.size() - 1 && A[k + 1][l] == '1') connect(A, k + 1, l);
        if(l < A[0].size() - 1 && A[k][l + 1] == '1') connect(A, k , l + 1);
        if(k > 0 && A[k - 1][l] == '1') connect(A, k - 1, l);
        if(l > 0 && A[k][l - 1] == '1') connect(A, k , l - 1);
    }
};