Question

419. 甲板上的战舰

English Version

题目描述

给你一个大小为 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 的值来解决这个问题吗？

解法

方法一

class Solution:
  def countBattleships(self, board: List[List[str]]) -> int:
    m, n = len(board), len(board[0])
    ans = 0
    for i in range(m):
      for j in range(n):
        if board[i][j] == '.':
          continue
        if i > 0 and board[i - 1][j] == 'X':
          continue
        if j > 0 and board[i][j - 1] == 'X':
          continue
        ans += 1
    return ans

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

class Solution {
public:
  int countBattleships(vector<vector<char>>& board) {
    int m = board.size(), n = board[0].size();
    int ans = 0;
    for (int i = 0; i < m; ++i) {
      for (int j = 0; j < n; ++j) {
        if (board[i][j] == '.') continue;
        if (i > 0 && board[i - 1][j] == 'X') continue;
        if (j > 0 && board[i][j - 1] == 'X') continue;
        ++ans;
      }
    }
    return ans;
  }
};

func countBattleships(board [][]byte) int {
  m, n := len(board), len(board[0])
  ans := 0
  for i := 0; i < m; i++ {
    for j := 0; j < n; j++ {
      if board[i][j] == '.' {
        continue
      }
      if i > 0 && board[i-1][j] == 'X' {
        continue
      }
      if j > 0 && board[i][j-1] == 'X' {
        continue
      }
      ans++
    }
  }
  return ans
}