題目鏈接:https://leetcode-cn.com/problems/number-of-islands/
題目描述
給你一個由 ‘1’(陸地)和 ‘0’(水)組成的的二維網格,請你計算網格中島嶼的數量。
島嶼總是被水包圍,並且每座島嶼只能由水平方向和/或豎直方向上相鄰的陸地連接形成。
此外,你可以假設該網格的四條邊均被水包圍。
第一次編輯代碼:
(他寫垃圾算法一直可以的)
class Solution {
public int numIslands(char[][] grid) {
if(grid == null || grid.length == 0)
return 0;
int row = grid.length, col = grid[0].length;
int ans = 0;
for(int i = 0; i < row; i++)
for(int j = 0; j < col; j++){
if(grid[i][j] == '1'){
dfs(i, j, grid, row, col);
ans++;
}
}
return ans;
}
public void dfs(int i, int j, char[][] grid, int row, int col){
grid[i][j] = '2';
if(i-1 >= 0 && grid[i-1][j] == '1')
dfs(i-1, j, grid, row, col);
if(i+1 < row && grid[i+1][j] == '1')
dfs(i+1, j, grid, row, col);
if(j-1 >= 0 && grid[i][j-1] == '1')
dfs(i, j-1, grid, row, col);
if(j+1 < col && grid[i][j+1] == '1')
dfs(i, j+1, grid, row, col);
}
}
反思
太蠢了,判斷條件可以提出來不用重複寫。
第二次編輯代碼:
class Solution {
public int numIslands(char[][] grid) {
if(grid == null || grid.length == 0)
return 0;
int row = grid.length, col = grid[0].length;
int ans = 0;
for(int i = 0; i < row; i++)
for(int j = 0; j < col; j++){
if(grid[i][j] == '1'){
dfs(i, j, grid, row, col);
ans++;
}
}
return ans;
}
public void dfs(int i, int j, char[][] grid, int row, int col){
if(i < 0 || i >= row || j < 0 || j >= col || grid[i][j] != '1')
return;
grid[i][j] = '2';
dfs(i-1, j, grid, row, col);
dfs(i+1, j, grid, row, col);
dfs(i, j-1, grid, row, col);
dfs(i, j+1, grid, row, col);
}
}