我正在研究Number of Groups (or "islands") of 1's in a Matrix的各种解决方案,而以下是明确的&简洁的Java解决方案看起来方向正确,对我来说看起来也不完整:
/*
* Given a matrix of 0's and 1's,
* find the number of groups of 1's in the matrix.
*
* A group of 1's is defined as all ADJACENT 1's
* vertically or horizontally but not diagonally.
*/
public class Islands {
/**
* main entry point
*/
public static void main(String[] args) {
int[][] A = new int[4][4];
int totalNumGroups = 0;
int curCnt = 0;
/*
* Initialize 2-dimensional array with 1's and 0's (randomly!)
* For testing/verification purpose only
*/
for(int x=0; x<A.length; x++) {
for(int y=0; y<A[x].length; y++) {
A[x][y] = (int) Math.round(Math.random());
System.out.print(A[x][y] + " ");
}
System.out.println(" ");
}
/*
* The crux of the solution: iterate through all (x,y):
* If encountered a 1,
* reset current count and
* increase total number of groups by what clean_block returns.
*/
for(int x=0; x<A.length; x++) {
for(int y=0; y<A[x].length; y++) {
if (A[x][y] == 1) {
curCnt = 0;
totalNumGroups = totalNumGroups + cleanBlock(A, x,y, curCnt);
}
// else (0), keep curCnt and totalNumGroups as are.
}
}
System.out.println("\nTotal # of groups: " + totalNumGroups);
}
/*
* Recursively clean found 1 and its adjacent 1's.
*/
public static int cleanBlock(int[][] A, int x, int y, int cnt) {
A[x][y] = 0;
if (inMatrix(x-1,y ,A.length,A[0].length) == 1 && A[x-1][y] == 1) {
cleanBlock(A, x-1,y ,cnt);
cnt = 1;
}
if (inMatrix(x+1,y ,A.length,A[0].length) == 1 && A[x+1][y] == 1) {
cleanBlock(A, x+1,y ,cnt);
cnt = 1;
}
if (inMatrix(x,y-1 ,A.length,A[0].length) == 1 && A[x][y-1] == 1) {
cleanBlock(A, x,y-1 ,cnt);
cnt = 1;
}
if (inMatrix(x,y+1 ,A.length,A[0].length) == 1 && A[x][y+1] == 1) {
cleanBlock(A, x,y+1 ,cnt);
cnt = 1;
}
return cnt;
}
public static int inMatrix(int x, int y, int lenX, int lenY) {
if ( (x >= 0 && x <= (lenX-1)) && (y >= 0 && y <= (lenY-1)) )
return 1;
else
return 0;
}
}
这是因为它不计算单个1(被0包围)作为一个组。例如此4x4矩阵的输出仅产生一个组:
1 1 0 1
1 0 0 0
1 1 0 1
1 0 0 0
Total # of groups: 1
所以,我的问题是:被0包围的单个1被认为是一个群体吗?
答案 0 :(得分:1)
这是正确的,因为根据问题:
如果1垂直或存在1,则可形成1组 水平到相邻的1
所以在你的情况下,孤独的1不能算作一个群体,因为没有其他1个水平或垂直相邻。