쵼쥬
쵼쥬의 개발공부 TIL
쵼쥬
전체 방문자
오늘
어제
  • 분류 전체보기 (276)
    • 코딩테스트 (192)
      • [알고리즘] 알고리즘 정리 (7)
      • [백준] 코딩테스트 연습 (126)
      • [프로그래머스] 코딩테스트 연습 (59)
    • Spring (71)
      • [인프런] 스프링 핵심 원리- 기본편 (9)
      • [인프런] 스프링 MVC 1 (6)
      • [인프런] 스프링 MVC 2 (4)
      • [인프런] 실전! 스프링 부트와 JPA 활용1 (7)
      • [인프런] 실전! 스프링 부트와 JPA 활용2 (5)
      • [인프런] 실전! 스프링 데이터 JPA (7)
      • [인프런] 실전! Querydsl (7)
      • JWT (5)
      • [인프런] Spring Cloud (17)
      • [인프런] Spring Batch (4)
    • Java (6)
      • [Java8] 모던인자바액션 (4)
      • [부스트코스] 웹 백엔드 (2)
      • [패스트캠퍼스] JAVA STREAM (0)
    • CS (6)
      • 디자인 패턴과 프로그래밍 패터다임 (2)
      • 네트워크 (4)

블로그 메뉴

  • 홈

공지사항

인기 글

태그

  • jpa
  • 구현
  • 비트마스킹
  • 위클리 챌린지
  • 백준
  • 자바
  • MVC
  • 부스트코스
  • 코딩테스트
  • 스프링
  • 인프런
  • 백분
  • querydsl
  • 프로그래머스
  • 누적합
  • Spring Data JPA
  • 알고리즘
  • 타임리프
  • BFS
  • spring

최근 댓글

최근 글

티스토리

hELLO · Designed By 정상우.
쵼쥬

쵼쥬의 개발공부 TIL

탈옥 - 9376번
코딩테스트/[백준] 코딩테스트 연습

탈옥 - 9376번

2022. 3. 15. 18:00


내 코드

package com.company;

import java.io.*;
import java.util.*;

// 최대한 문이 겹치도록 해야함
// -> 죄수 두명이 만나서 같이 나가도록 해서 문에 겹치도록 함
// 만나서 밖으로 나감 -> 반대로 하면 밖에서 죄수 두명이 만나는 지점까지 이동
// -> 상근이가 밖에서 안으로 들어온다고 생각해서 3명이 만나는 지점을 찾아줌
public class Main {
    static int[][] d = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
    static int h, w, answer;
    static int[][][] route;
    static int[][] arr;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;

        int T = Integer.parseInt(br.readLine());

        for (int tc = 0; tc < T; tc++) {
            st = new StringTokenizer(br.readLine());
            h = Integer.parseInt(st.nextToken());
            w = Integer.parseInt(st.nextToken());

            arr = new int[h + 2][w + 2];        // 감옥
            route = new int[3][h + 2][w + 2];   // 죄수와 상근이가각 좌표마다 이동할때 만나는 문의 갯수 저장
            int[] prisonerA = new int[3];   // 죄수 위치
            int[] prisonerB = new int[3];   // 죄수 위치
            int[] sang = {0, 0, 0};         // 상근이?
            answer = Integer.MAX_VALUE;

            // 최소 경로를 찾기위해 일단 최대로 설정
            for (int k = 0; k < 3; k++) {
                for (int i = 0; i < h + 2; i++) {
                    Arrays.fill(route[k][i], 10000);
                }
            }

            route[0][0][0] = 0; // 상근이 시작하는 위치

            for (int i = 1; i <= h; i++) {
                String s = br.readLine();
                for (int j = 1; j <= w; j++) {
                    char c = s.charAt(j - 1);
                    if (c == '*') {         // 벽은 -1로 설정
                        arr[i][j] = -1;
                    } else if (c == '#') {  // 문은 1로 설정
                        arr[i][j] = 1;
                    } else if (c == '$') {  // 죄수 위치 설정
                        if (prisonerA[0] == 0) {
                            prisonerA[0] = i;
                            prisonerA[1] = j;
                            route[1][i][j] = 0;
                        } else {
                            prisonerB[0] = i;
                            prisonerB[1] = j;
                            route[2][i][j] = 0;
                        }
                    }
                }
            }

            bfs(0, sang);   // 상근이의 최소 경로 구함
            bfs(1, prisonerA);  // 죄수들의 최소 경로 구함
            bfs(2, prisonerB);

            System.out.println(answer);
        }
    }

    static void bfs(int x, int[] man) {

        // 문의 갯수가 작은 순서대로 정렬한 우선 순위 큐
        PriorityQueue<int[]> q = new PriorityQueue<>((o1, o2) -> o1[2] - o2[2]);

        q.add(man);
        boolean[][] visited = new boolean[h + 2][w + 2];

        while (!q.isEmpty()) {
            int[] pos = q.poll();
            visited[pos[0]][pos[1]] = true;
            // 2두번째 죄수 탐색이면 정답 찾음
            if (x == 2) {
                int count = route[0][pos[0]][pos[1]] + route[1][pos[0]][pos[1]] + route[2][pos[0]][pos[1]];

                answer = Math.min(answer, arr[pos[0]][pos[1]] == 1 ? count - 2 : count);
            }

            for (int i = 0; i < 4; i++) {
                int nextX = pos[0] + d[i][0];
                int nextY = pos[1] + d[i][1];

                // 다음 위치가 벽이 아니면 탐색
                if (nextX >= 0 && nextY >= 0 && nextX < h + 2 && nextY < w + 2
                        && !visited[nextX][nextY] && arr[nextX][nextY] != -1) {
                    // 다음 위치가 문이면 +1
                    route[x][nextX][nextY] = arr[nextX][nextY] == 1 ? route[x][pos[0]][pos[1]] + 1 : route[x][pos[0]][pos[1]];
                    visited[nextX][nextY] = true;
                    q.add(new int[]{nextX, nextY, route[x][nextX][nextY]});
                }
            }
        }
    }
}

 

'코딩테스트 > [백준] 코딩테스트 연습' 카테고리의 다른 글

나머지 합 - 10986번  (0) 2022.03.18
로봇 - 1726번  (0) 2022.03.17
욕심쟁이 판다 - 1937번  (0) 2022.03.14
가스관 - 2931번  (0) 2022.03.13
구슬 탈출2 - 13460번  (0) 2022.03.10
    '코딩테스트/[백준] 코딩테스트 연습' 카테고리의 다른 글
    • 나머지 합 - 10986번
    • 로봇 - 1726번
    • 욕심쟁이 판다 - 1937번
    • 가스관 - 2931번
    쵼쥬
    쵼쥬

    티스토리툴바