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};};