一个矩阵中只有0和1两种值,每个位值都可以和自己的上、下、左、右四个位值相连,如果有一片1连在一起,这个部分叫做一个岛,求一个矩阵中有多少个岛?
举例:
0 0 1 0 1 0
1 1 1 0 1 0
1 0 0 1 0 0
0 0 0 0 0 0
这个矩阵中有三个岛。
先来一个思路比较简单的方法:
遍历整个地图,每次遇到1,就把1改成2,把连着一片都改完了就把岛屿数加一。
class Island{
public:
void infect(vector<vector<int>> &m,int i,int j,int N,int M){
if(i<0||i>=N||j<0||j>=M||m[i][j]!=1)
return;
m[i][j]=2;
infect(m,i+1,j,N,M);
infect(m,i-1,j,N,M);
infect(m,i,j+1,N,M);
infect(m,i,i-1,N,M);
}
int countIsland(vector<vector <int>> &m){
if(m.empty())
return 0;
int N=m.size();
int M=m[0].size();
int res=0;
for(int i=0;i<N;i++){
for(int j=0;i<M;j++){
if(m[i][j]==1){
res++;
infect(m, i, j, N, M);
}
}
}
return res;
}
};
上面这个方法,用的是单cpu,单内存的方法。
还可以把这个地图进行划分,各自统计每一块的岛屿个数,然后以合适的逻辑,统计结果进行合并。
到底用什么样的逻辑进行合并呢?
查并集
假如我们是把这样一份地图划分成两份
每一份都用infect的方法进行标记,同一个岛屿里的各个点放在同一个集合里,拥有同一个头节点,整个过程需要留下的信息,就是每一块的岛屿个数,以及,标记每个1节点所属的集合。
把左右两边的岛屿数加起来,下面这张图,左右两边加起来,是4,但合起来其实只有两座岛屿,所以下面开始去重的工作。
比较分界线两边的节点,第一行两边属于不同的集合A和C,那么就在岛屿数减1,岛屿数变成3,然后把A和C挂在同一个集合下面,然后来到下一行,所有两边都在同一个集合上,岛屿数不变,跳下一行,一边是0,岛屿数不变,跳下一行,B和C两个不同的集合,岛屿数减1,岛屿数为2,再下一行岛屿数不变,所以整个分界线检查结束后,岛屿数为2,结果正确
分成更多块的统计方法是相同的,只需要在四个边界上进行查重