给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ,返回在甲板 board 上放置的 战舰 的数量。
战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
输入:board = [[“X”,“.”,“.”,“X”],[“.”,“.”,“.”,“X”],[“.”,“.”,“.”,“X”]]。
输出:2。
力扣419. 甲板上的战舰。
并查集或者岛问题都行,但这不是最优解。
数战舰的左上角,统计左上角的点的个数就行。
时间复杂度:O(N**2)。
代码用rust编写。代码如下:
fn main() {
let m: Vec<Vec<char>> = vec![
vec!['X', '.', '.', 'X'],
vec!['.', '.', '.', 'X'],
vec!['.', '.', '.', 'X'],
];
let ans = count_battleships(m);
println!("ans = {}", ans);
}
fn count_battleships(m: Vec<Vec<char>>) -> i32 {
let mut ans: i32 = 0;
for i in 0..m.len() {
for j in 0..m[0].len() {
if m[i][j] == 'X' && (i == 0 || m[i - 1][j] != 'X') && (j == 0 || m[i][j - 1] != 'X') {
ans += 1;
}
}
}
ans
}
执行结果如下: