Java 20125 쿠키의 신체 측정

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

20125번: 쿠키의 신체 측정

풀이 :

1. 쿠키의 머리를 찾은 뒤 머리의 바로 한칸 아래가 심장이므로 hX, hY  변수에 심장에 위치를 업데이트한다.

2. 쿠키의 왼팔, 오른팔, 허리 순으로 dfs 탐색을 진행하여 길이를 구해준다. 허리의 경우에 허리의 끝 양쪽에 다리가 있으므로 해당 좌표를 static 변수에 업데이트 해준다.

3. bX, bY 변수는 허리의 끝 지점이다. 양쪽 다리에 대해서 dfs 탐색을 진행하여 양다리의 길이를 구한다.

 

위의 로직대로 문제를 해결하였습니다.

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;

public class Main {
    static int[] dx = {0, 0, 1};
    static int[] dy = {-1, 1, 0};
    static int[][] map;
    static int n, bX, bY;
    static boolean check = false;
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        map = new int[n][n];

        boolean findHead = false;
        int hX = 0;
        int hY = 0;

        for (int i = 0; i < n; i++) {
            String str = br.readLine();
            for (int j = 0; j < n; j++) {
                if (str.charAt(j) == '*') {
                    map[i][j] = 1;
                    if (!findHead) {
                        hX = i + 1;
                        hY = j;
                        findHead = true;
                    }
                } else {
                    map[i][j] = 0;
                }
            }
        }
        sb.append(hX+1).append(" ").append(hY+1).append("\n");

        for (int i = 0; i < 3; i++) {
            dfs(hX+dx[i], hY+dy[i], 0, i);
        }

        dfs(bX+1, bY-1, 1, 2);
        dfs(bX+1, bY+1, 1, 2);

        System.out.println(sb);
    }

    static void dfs(int x, int y, int cnt, int dir) {
        if (x < 0 || y < 0 || x >= n || y >= n || map[x][y] != 1) {
            sb.append(cnt).append(" ");
            if(dir == 2 && !check){
                bX = x;
                bY = y;
                check = true;
            }
            return;
        } else {
            dfs(x + dx[dir], y + dy[dir], cnt + 1, dir);
        }
    }
}