题目描述
给定一个二维的甲板, 请计算其中有多少艘战舰。 战舰用 'X’表示,空位用 '.'表示。 你需要遵守以下规则:
给你一个有效的甲板,仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成,或者 Nx1 (N 行, 1 列)组成,其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
示例 :
X…X
…X
…X
在上面的甲板中有2艘战舰。
无效样例 :
…X
XXXX
…X
你不会收到这样的无效甲板 - 因为战舰之间至少会有一个空位将它们分开。
进阶:
你可以用一次扫描算法,只使用O(1)额外空间,并且不修改甲板的值来解决这个问题吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/battleships-in-a-board
解法1
1.这里首先遍历board的整个数组,之后若是遇到x的则检查(x+1,y)和(x,y+1)若是有x则继续遍历
2.找到对应的x则变为.,且数量加加,对于当前x的后面可以探测两个方向,一个探测横向,一个探测纵向。这里要注意越界。
3.记录x的最开始位置,该x遍历完成之后,从起始x的位置遍历。
缺点与不足:
1.这里不止一次扫描,并且修改了甲板的值
int countBattleships(vector<vector<char>>& board) {
int count = 0;
int row = board.size();
int column = board[0].size();
int ind;
//这里遍历每个board按照行优先遍历矩阵。
for(int i = 0; i < row; i++) {
for(int j = 0; j < column; j++) {
//如果当前元素是'X',那么自动加加,将当前元素置位'.'
if(board[i][j] == 'X') {
count++;
board[i][j] = '.';
//检查是否行向量中有对应的X
//同时检查范围是否越界
ind = i + 1;
//范围合法且当前元素是'X'
while(ind < row && board[ind][j] == 'X') {
board[ind][j] = '.';
ind++;
}
//检查列向量中有对应的X
ind = j + 1;
while(ind < column && board[i][ind] == 'X') {
board[i][ind] = '.';
ind++;
}
}
}
}
return count;
}
只遍历舰头,若不是舰头,则加加
判断是不是舰头,看(x-1,y)和(x,y-1)是否有对应的元素
这次只遍历一次board,且不修改值,而且不需要其他的额外空间
只遍历舰头位置,也就是左边和上边如果有’X’,那么就不是舰头
int countBattleships(vector<vector<char>>& board) {
int row = board.size();
int column = board[0].size();
int count = 0;
for(int i = 0; i < row; i++) {
for(int j = 0; j < column; j++) {
//首先检查上面是否存在,若存在且不等于'X'
//之后在检查左边是否存在,若存在且不等于'X'
//那么count++
//如果当前的元素是'X',那么再继续判断是否是舰头,否则直接跳过
if(board[i][j] != 'X') {
continue;
}
//如果当前元素是舰头,检查上一个是否存在,存在不等于'X'在继续
if(i - 1 >= 0 && i - 1 < row) {
if(board[i - 1][j] == 'X') {
continue;
}
}
//如果当前元素是舰头,检查左边一个是否存在,存在不等于'X'在继续
if(j - 1 >= 0 && j - 1 < column) {
if(board[i][j - 1] == 'X') {
continue;
}
}
//执行到这里的原因是当前元素是'X',且上边和左边都不是'X',那么就是舰头
count++;
}
}
return count;
}
在一位题解中看到这样的题解,感觉也很好。很简洁
if (board[i][j] == 'X') {
char a = i==0?'.':board[i-1][j];
char b = j==0?'.':board[i][j-1];
if (a!='X'&&b!='X') {
ans++;
}
}
int main() {
vector<vector<char>> board(3, vector<char>(4));
board[0][0] = 'X';
board[0][1] = '.';
board[0][2] = '.';
board[0][3] = 'X';
board[1][0] = '.';
board[1][1] = '.';
board[1][2] = '.';
board[1][3] = 'X';
board[2][0] = '.';
board[2][1] = '.';
board[2][2] = '.';
board[2][3] = 'X';
int ret = countBattleships(board);
cout << ret << endl;
return 0;
}