2017.1.10 算法测试题集 - 1003 - 1的块数

Problem

给定一个0-1矩阵matrix，相邻的1视为“同一块”，问矩阵matrix有多少块1？

Example

[1, 0, 0]
[0, 1, 0]
[0, 0, 1]
Return 3

[1, 0, 0, 1]
[0, 1, 1, 0]
[0, 0, 0, 1]
Return 4;

---

Algorithm

通过深度优先搜索来解决。

遍历矩阵，若访问到1，则将这个1以及与它相连的所有1都置为0，同时块数加一。重复以上过程直到遍历完成。此时矩阵所有元素都变成了0。

具体一些细节要注意的是，将相连的1修改为0的过程需要用递归实现。寻找周围的1的时候要注意是否越界（是否在矩阵界内）。

考试的时候还算比较快找到思路，只是实现的时候细节不够注意。对于这类题目关键是细心，考虑全面。

代码如下。

class Solution {
public: 
    bool isin(int i,int j,int m,int n){
        if(i<0||i>m-1||j<0||j>n-1) return false;
        else return true;
    }

    void setZero(int i,int j,int m,int n,vector<vector<char> >& matrix){
        matrix[i][j]='0';
        for(int k=0;k<4;k++){
            if(isin(i+dy[k],j+dx[k],m,n)&&matrix[i+dy[k]][j+dx[k]]!='0'){
                setZero(i+dy[k],j+dx[k],m,n,matrix);
            }
        }
    }   

    int blocks(vector<vector<char> >& matrix){
        int m=matrix.size();
        if(m==0) return 0;
        int n=matrix[0].size();
        int sum=0;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(matrix[i][j]!='0'){
                    sum++;
                    setZero(i,j,m,n,matrix);
                }
            }
        }
        return sum;
    }
private:
    int dx[4]={0,0,-1,1};
    int dy[4]={1,-1,0,0};
};