[NeetCode-LeetCode] Max Area of Island JAVA

반응형

728x90

[NeetCode-LeetCode] Max Area of Island - Medium

---

접근

• dfs

---

풀이

2차원 배열이 주어지고 1로 이루어진 구간의 최대 넓이를 구하는 문제이다. 비슷한 유형이 많은 문제이기 때문에 dfs를 통해 간단하게 해결해 주면 된다. 

    private static final int[] dx = {0, 0, 1, -1};
    private static final int[] dy = {1, -1, 0, 0};

    public int maxAreaOfIsland(int[][] grid) {
        int answer = 0;

        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[i].length; j++) {
                if (grid[i][j] == 1) {
                    grid[i][j] = 0;
                    answer = Math.max(answer, dfs(grid, i, j));
                }
            }
        }
        
        return answer;
    }

grid의 값이 1인 경우에 dfs를 실행하여 넓이를 구하게 된다. 

    private int dfs(int[][] grid, int x, int y) {
        int count = 1;

        for (int d = 0; d < 4; d++) {
            int nx = x + dx[d];
            int ny = y + dy[d];

            if (nx < 0 || nx >= grid.length || ny < 0 || ny >= grid[nx].length) {
                continue;
            }

            if (grid[nx][ny] == 0) {
                continue;
            }

            grid[nx][ny] = 0;
            count += dfs(grid, nx, ny);
        }

        return count;
    }

넓이를 구하는 dfs 코드이다. 다음 좌표를 구하고 해당 좌표의 값이 배열의 인덱스를 벗어나거나 0이라면 건너뛰고 1인 경우에만 탐색을 진행하게 된다. 

---

전체 코드

class Solution {
    private static final int[] dx = {0, 0, 1, -1};
    private static final int[] dy = {1, -1, 0, 0};

    public int maxAreaOfIsland(int[][] grid) {
        int answer = 0;

        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[i].length; j++) {
                if (grid[i][j] == 1) {
                    grid[i][j] = 0;
                    answer = Math.max(answer, dfs(grid, i, j));
                }
            }
        }
        
        return answer;
    }

    private int dfs(int[][] grid, int x, int y) {
        int count = 1;

        for (int d = 0; d < 4; d++) {
            int nx = x + dx[d];
            int ny = y + dy[d];

            if (nx < 0 || nx >= grid.length || ny < 0 || ny >= grid[nx].length) {
                continue;
            }

            if (grid[nx][ny] == 0) {
                continue;
            }

            grid[nx][ny] = 0;
            count += dfs(grid, nx, ny);
        }

        return count;
    }
}