给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 'X' 或者是一个空位 '.' ,返回在甲板 board 上放置的 战舰 的数量。
战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
示例 1:
输入:board = [["X",".",".","X"],[".",".",".","X"],[".",".",".","X"]]
输出:2
示例 2:
输入:board = [["."]]
输出:0
提示:
m == board.length
n == board[i].length
1 <= m, n <= 200
board[i][j] 是 '.' 或 'X'
进阶:你可以实现一次扫描算法,并只使用 O(1) 额外空间,并且不修改 board 的值来解决这个问题吗?
。
不相邻,左上往右下角进行扫描
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int countBattleships(vector<vector<char>>& board) {
int sum = 0;
for (int i = 0; i < board.size(); i++) {
for (int j = 0; j < board[i].size(); j++) {
if (board[i][j] == 'X') {
dealBattleShips(board, i, j, sum);
}
}
}
return sum;
}
void dealBattleShips(vector<vector<char>>& board, int x, int y, int& sum) {
for (int i = x + 1; i < board.size() && board[i][y] == 'X'; i++) {
board[i][y] = '.';
}
for (int j = y + 1; j < board[x].size() && board[x][j] == 'X'; j++) {
board[x][j] = '.';
}
sum++;
}
};
int main() {
vector<vector<char>> v = { {'X', '.', '.', 'X'},{'.', '.', '.', 'X'},{'.', '.', '.', 'X'} };
unique_ptr<Solution> p;
cout << p->countBattleships(v) << endl;
system("pause");
return 0;
}