BFS/DFS中等 LeetCode419. 甲板上的战舰

描述

给你一个大小为 m x n 的矩阵 board 表示甲板，其中，每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ，返回在甲板 board 上放置的 战舰 的数量。

战舰 只能水平或者垂直放置在 board 上。换句话说，战舰只能按 1 x k（1 行，k 列）或 k x 1（k 行，1 列）的形状建造，其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 （即没有相邻的战舰）。

分析

战舰与战舰是分离不接壤的，所以连在一起的’X’是一个战舰。
遇到’X’，统计累加，并把与其连接的’X’都标记，防止再次累加。
标记防止重复遍历的手段有：辅助数组标记、原数组改变值，两种手段。

class Solution {
    public int countBattleships(char[][] board) {
        int ans = 0;
        int m = board.length, n = board[0].length;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(board[i][j] != 'X'){
                    continue;
                }
                ans++;
                int k = i+1;
                int u = j+1;
                while(u < n && board[i][u] == 'X'){
                    board[i][u] = 'O';
                    u++;
                }
                while(k < m && board[k][j] == 'X'){
                    board[k][j] = 'O';
                    k++;
                }
                board[i][j] = 'O';
            }
        }
        return ans;
    }
}