正常来说,这种找一个图里面有多少个island的题目都是用什么dfs来做的。就是dfs走遍一个island或者battleship然后mark visited之类的。这题也是可以的。当然follow up就肯定不是这样了。这题给了一个很给力的条件,就是所有的battleship都是一条线。要不就是直的,要不就是横的,并且任意两条船之间,都有空位,不能相邻。所以这个时候,其实我们只需要找到每条battleship的头部就可以了,因为battleship either 1 * n或者n * 1,所以,每条battleship的头部,要不就是最上边的那个点,或者最左边的一个点。所以,只要判断一个点为X,并且在它之上或者之左没有别的点,判断就成立了。其余x点都是battleship的中间点,不需要计算入内。根据这个算法,可以得到代码如下:
public int countBattleships(char[][] board) {
int count = 0;
for (int i = 0; i < board.length; i++) {
for (int j = 0; j < board[0].length; j++) {
if (board[i][j] == 'X') {
boolean noTop = (i == 0) || (board[i - 1][j] != 'X');
boolean noLeft = (j == 0) || (board[i][j - 1] != 'X');
if (noTop && noLeft) count++;
}
}
}
return count;
}
顺便说一下,只要符合任意两条船都不相邻的条件,battleship是m * n的,也就是是一个方形的,这个算法也是可以成立的。