题目:
对于一个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.
思路:DFS
代码:
class Solution {
public:
int countConnectedOnes(vector<vector<char>>& A) {
if(A.empty())
return 0;
m=A.size();
n=A[0].size();
int count=0;
for(int i=0;i<m;i++) {
for (int j=0;j<n;j++) {
if(A[i][j]=='1') {
dfs(i,j,A);
count++;
}
}
}
return count;
}
private:
int m,n;
void dfs(int i,int j,vector<vector<char>>& A) {
if(i<0||i>=m||j<0||j>=n)
return;
if(A[i][j]!='1')
return;
else {
A[i][j]='2';
dfs(i-1,j,A);
dfs(i+1,j,A);
dfs(i,j-1,A);
dfs(i,j+1,A);
}
}
};