题目:
给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ,返回在甲板 board 上放置的 战舰 的数量。战舰只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
示例:
输入:board = [[“X”,".",".",“X”],[".",".",".",“X”],[".",".",".",“X”]]
输出:2
解题思路:
使用循环遍历加更改的方法,即循环遍历加班向量board中的元素,当遍历到一个元素board[i][j]==‘X’时,证明该元素为战舰的一部分,然后将该点设定为’.’,代表已经遍历过,无需再遍历,于此同时,基于该点,分别在横向和纵向上延申遍历,并将’X’ 变为’.’,思想跟上述相同。只在横向和纵向上两个方向延申遍历的原因是因为题目中给出”战舰只能水平或者垂直放置在 board 上“。
Code
class Solution {
public:
int countBattleships(vector<vector<char>>& board) {
int num_row = board.size();
int num_col = board[0].size();
int ans = 0;
for(int i = 0;i< num_row;i++){
for (int j = 0;j< num_col;j++){
if (board[i][j] == 'X'){
board[i][j] = '.';
for (int k = i+1;k < num_row && board[k][j] == 'X';k++)
board[k][j] = '.';
for (int k = j+1;k < num_col && board[i][k] == 'X';k++)
board[i][k] = '.';
ans++;
}
}
}
return ans;
}
};