[LeetCode]419 Battleships in a Board（暴力，dfs）

题目链接：https://leetcode.com/problems/battleships-in-a-board/?tab=Description

题意：问平面上有多少条船，船是1*n或者n*1的，并且不相交。

最简单的想法就是直接dfs，但是看到题目里有一个思考，那就是不修改平面并且空间O(1)的one-pass。

这时候就想到应该是遍历所有船的左上角的端点，统计就行了。很简单。

public class Solution {
    public int countBattleships(char[][] board) {
        int ret = 0;
        for (int i = 0; i < board.length; i++) {
            for (int j = 0; j < board[0].length; j++) {
                if (board[i][j] == 'X' && (i == 0 || board[i-1][j] == '.') && (j == 0 || board[i][j-1] == '.')) {
                    ret++;
                }
            }
        }
        return ret;
    }
}

 

DFS的代码

class Solution {
public:
    const int dx[5] = {0, 0, 1, -1};
    const int dy[5] = {1, -1, 0, 0};
    int ret;
    int ok(vector<vector<char>>& board, int x, int y) {
        return x < board.size() && y < board[0].size();
    }

    void dfs(vector<vector<char>>& board, int x, int y) {
        board[x][y] = '.';
        for(int i = 0; i < 4; i++) {
            int xx = x + dx[i];
            int yy = y + dy[i];
            if(ok(board, xx, yy) && board[xx][yy] == 'X') {
                dfs(board, xx, yy);
            }
        }
    }

  int countBattleships(vector<vector<char>>& board) {
      ret = 0;
      for(int i = 0; i < board.size(); i++) {
          for(int j = 0; j < board[i].size(); j++) {
              if(board[i][j] == 'X') {
                  dfs(board, i, j);
                  ret++;
              }
          }
      }
      return ret;
  }
};