1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
| class Solution { int[][] move = {{0,1}, {0,-1}, {1,0}, {-1,0}}; int area = 0; int mark = 1;
public int largestIsland(int[][] grid) { int n = grid.length; int m = grid[0].length; Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < n; i++){ for(int j = 0; j < m; j++){ if(grid[i][j] == 1){ area = 0; mark ++; dfs(grid, i, j, mark); map.put(mark, area); } } }
int max_sum = 0; Set<Integer> set = new HashSet<>(); for(int i = 0; i< n; i++){ for(int j = 0; j < m; j++){ if(grid[i][j] == 0){ set.clear(); int sum = 1; for(int k = 0; k < 4; k++){ int next_i = i + move[k][0]; int next_j = j + move[k][1]; if(next_i < 0 || next_j < 0 || next_i >= n || next_j >= m || grid[next_i][next_j] == 0){ continue; } int num = grid[next_i][next_j]; if(set.contains(num)){ sum += map.get(num); set.add(num); } } max_sum = Math.max(max_sum, sum); }
} } return max_sum == 0 ? m * n : max_sum; }
public void dfs(int[][] grid, int x, int y, int mark){ if(x < 0 || y < 0 || x >= grid.length || y >= grid[0].length){ return; } if(grid[x][y] == 0 || grid[x][y] > 1){ return; } grid[x][y] = mark; area ++; dfs(grid, x + 1, y, mark); dfs(grid, x , y + 1, mark); fs(grid, x - 1, y, mark); dfs(grid, x, y - 1, mark); } }
|