‡ CODING TEST STUDY ‡/º 백준

[백준 | Java Silver I] (#1743) 음식물 피하기

Trudy | 송연 2024. 7. 16. 15:24

문제

https://www.acmicpc.net/problem/1743


접근

대표적인 DFS 문제인데, 음식물의 길이를 구해야 했던 문제

 

처음에는 dfs의 depth로 음식물의 길이를 구하려고 했는데, 잘못된 접근이 이었다. 

# . . .
. # # .
# # . .

 

위와 같은 경우 (2,2)에서 재귀가 시작되면 depth는 (2,2), (2,3) 으로 가는 경우 2, (2,2),(2,3),(1,3) 으로 가는 경우 3 해서 최대 depth는 3이 된다. 

 

따라서 이렇게는 음식물의 길이 4를 구할 수 없었다. 

 

 

첫번째 제출 - 실패 : dfs의 depth로 길이를 구하려고 잘못 접근함

package week11.baek.july16;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.StringTokenizer;

public class S1743 {
    static int[][] map;
    static boolean[][] visited;
    static int N, M;
    static int[] dx = {-1, 1, 0, 0};
    static int[] dy = {0, 0, -1, 1};
    static int max;

    public static void dfs(int x, int y, int depth){
        visited[x][y] = true;

        //depth의 max 값을 갱신
        if(max < depth) max = depth;

        int count = 1;

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

            if(nx < 0 || nx >= N || ny < 0 || ny >= M) continue;

            if(visited[nx][ny] == false && map[nx][ny] == 1 ){
                dfs(nx, ny, depth+1);
            }
        }

    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        int K = Integer.parseInt(st.nextToken());

        map = new int[N][M];
        for (int i = 0; i < K; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken()) - 1;
            int y = Integer.parseInt(st.nextToken()) - 1;
            map[x][y] = 1;
        }

        visited = new boolean[N][M];
        max = 0;

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if(map[i][j] == 1 && visited[i][j] == false) {
                    dfs(i, j, 1);
                }
            }
        }

        System.out.println(max);

    }
}

 

최종 제출

 

count를 첫 dfs 시작할 때 0으로 초기화하고 재귀로 dfs가 호출될때마다 count+1이 되도록 했다. 

그리고 max를 둬서, count가 가장 큰 값으로 갱신되도록 구현했다. 

package week11.baek.july16;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.StringTokenizer;

public class S1743 {
    static int[][] map;
    static boolean[][] visited;
    static int N, M;
    static int[] dx = {-1, 1, 0, 0};
    static int[] dy = {0, 0, -1, 1};
    static int max;
    static int count;

    public static void dfs(int x, int y, int depth){
        visited[x][y] = true;

        count += 1;

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

            if(nx < 0 || nx >= N || ny < 0 || ny >= M) continue;

            if(visited[nx][ny] == false && map[nx][ny] == 1 ){
                dfs(nx, ny, depth+1);
            }
        }

    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        int K = Integer.parseInt(st.nextToken());

        map = new int[N][M];
        for (int i = 0; i < K; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken()) - 1;
            int y = Integer.parseInt(st.nextToken()) - 1;
            map[x][y] = 1;
        }

        visited = new boolean[N][M];
        max = Integer.MIN_VALUE;

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if(map[i][j] == 1 && visited[i][j] == false) {
                    count = 0;
                    dfs(i, j, 1);
                    if(count > max) max = count;
                }
            }
        }

        System.out.println(max);

    }
}