419. 甲板上的战舰
给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ,返回在甲板 board 上放置的 战舰 的数量。
战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
class Solution {
int x_change[]={0,0,-1,1};
int y_change[]={1,-1,0,0};
public int countBattleships(char[][] board) {
int result=0;
int temp[]={0,0};
Stack<int[]> stack=new Stack<>();
int m=board.length;
int n=board[0].length;
int visit[][]=new int[m][n];
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(visit[i][j]==0&&board[i][j]=='X'){
result++;
temp[0]=i;
temp[1]=j;
stack.add(temp);
visit[i][j]=1;
while(!stack.isEmpty()){
int index[]=stack.pop();
int x=index[0];
int y=index[1];
for(int k=0;k<4;k++){
int x_new=x+x_change[k];
int y_new=y+y_change[k];
if(x_new>=0&&x_new<m&&y_new>=0&&y_new<n&&board[x_new][y_new]=='X'&&visit[x_new][y_new]==0){
int new_temp[]={x_new,y_new};
stack.add(new_temp);
visit[x_new][y_new]=1;
}
}
}
}
}
}
return result;
}
}