LeetCode 419 Java实现

LeetCode419

题目

给定一个二维的甲板， 请计算其中有多少艘战舰。 战舰用 'X’表示，空位用 '.'表示。 你需要遵守以下规则：

给你一个有效的甲板，仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成，或者 Nx1 (N 行, 1 列)组成，其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
示例 :

X…X
…X
…X
在上面的甲板中有2艘战舰。

无效样例 :

…X
XXXX
…X
你不会收到这样的无效甲板 - 因为战舰之间至少会有一个空位将它们分开。

代码

看题意我们知道两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
所以我们可以随心所欲的判断某一个点只要四周有X我们就把这个值换为.

class Solution {
    public int countBattleships(char[][] board) {
        int x = board.length;
        int y = board[0].length;
        int count = 0;
        for (int i = 0; i < x; i++)
        //如果上下左右有存在X,则这个X会被修改为.
        {
            for(int j = 0; j < y; j++){
                if(i - 1 >= 0 && board[i-1][j] == 'X')
                    board[i][j] = '.';
                if(i + 1 < board.length && board[i+1][j] == 'X')
                    board[i][j] = '.';
                if(j - 1 >= 0 && board[i][j-1] == 'X')
                    board[i][j] = '.';
                if(j + 1 < board[0].length && board[i][j+1] == 'X')
                    board[i][j] = '.';
            }
        }
        for (int i = 0; i < x; i++)
        //遍历计算有多少个X
        {
            for(int j = 0; j < y; j++){
                if(board[i][j] == 'X'){
                    count++;
                }
            }
        }
        return count;
    }
}