题目
给你一个大小为 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
解题
方法一:DFS计算连通域
由于题目中告知,任何两个战舰之间都有水平或者垂直空格,因此可以直接朝4个方向dfs,来计算得连通域
class Solution { public: int countBattleships(vector<vector<char>>& board) { int res=0; int m=board.size(); int n=board[0].size(); vector<vector<bool>> visited(m,vector<bool>(n,false)); function<void(int,int,int)> dfs=[&](int row,int col,int flag){ if(row<0||row>=m||col<0||col>=n||visited[row][col]) return; if(board[row][col]=='X'){ visited[row][col]=true; dfs(row+1,col,1); dfs(row-1,col,1); dfs(row,col+1,1); dfs(row,col-1,1); if(flag==0) res++; } }; for(int i=0;i<m;i++){ for(int j=0;j<n;j++){ dfs(i,j,0); } } return res; } };
方法二:一次扫描
class Solution { public: int countBattleships(vector<vector<char>>& board) { int m=board.size(); int n=board[0].size(); int res=0; for(int i=0;i<m;i++){ for(int j=0;j<n;j++){ if(i>0&&board[i-1][j]=='X') continue; if(j>0&&board[i][j-1]=='X') continue; if(board[i][j]=='X') res++; } } return res; } };