题目
给定一个二维的甲板, 请计算其中有多少艘战舰。 战舰用 'X’表示,空位用 '.'表示。 你需要遵守以下规则:
给你一个有效的甲板,仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成,或者 Nx1 (N 行, 1 列)组成,其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
你可以用一次扫描算法,只使用O(1)额外空间,并且不修改甲板的值来解决这个问题吗?
思路
我自己的思路还是去进行一个深搜或者广搜,这样子虽然可以完成,但是解决不了只使用一次扫描算法,O(1)的额外空间,并且还不修改甲板的值。
题解的解法,只统计战舰的左上角。因为战舰都是 1 行, N 列 或者 N 行, 1 列 的。所以,每次我们只要统计到 ‘X’,并且 'X’位于战舰的左上角即可。
代码
class Solution {
public int countBattleships(char[][] board) {
int row = board.length;
int column = board[0].length;
int ans=0;
for(int i=0;i<row;++i)
{
for(int j=0;j<column;++j)
{
// 当前是战舰
if(board[i][j]=='X')
{
// 可能是在第一行或者第一列,并且 左边和上边不是X,即是一个独立的战舰!
if((i==0||board[i-1][j]!='X')&&(j==0||board[i][j-1]!='X'))
++ans;
}
}
}
return ans;
}
}
结果
很巧妙的一个思路!