반응형
https://www.acmicpc.net/problem/16236
package com.ji.beakjoon.bfs;
import java.util.PriorityQueue;
import java.util.Scanner;
public class BabyShark {
static int[] dy = {-1, 0, 0, 1};
static int[] dx = {0, -1, 1, 0};
static int[][] map;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int N = sc.nextInt();
map = new int[N][N];
int[] cur = null;
for (int i = 0; i < N; i++)
for (int j = 0; j < N; j++) {
map[i][j] = sc.nextInt();
if (map[i][j] == 9) {
cur = new int[]{i, j};
map[i][j] = 0;
}
}
int size = 2;
int eat = 0; // 먹은 물고기 수
int move = 0; // 움직인 총 거리
while (true) {
// 거리가 가깝거나, 가장 위에 있거나, 가장 왼쪽에 있는 물고기 순
// 일반적인 Queue를 사용하면 상,좌,우,하 탐색하다가 규칙에 어긋나게됨
// 우선순위를 큐를 사용하여 이를 해결해야함
PriorityQueue<int[]> que = new PriorityQueue<>((o1, o2) ->
o1[2] != o2[2] ? Integer.compare(o1[2], o2[2])
: (o1[0] != o2[0] ? Integer.compare(o1[0], o2[0])
: Integer.compare(o1[1], o2[1]))
);
boolean[][] visit = new boolean[N][N];
que.add(new int[]{cur[0], cur[1], 0}); // y좌표, x좌표, 이동한 거리
visit[cur[0]][cur[1]] = true;
boolean ck = false; // 상어가 먹이를 먹었는지 체크할 변수
while (!que.isEmpty()) {
cur = que.poll();
if (map[cur[0]][cur[1]] != 0 && map[cur[0]][cur[1]] < size) { // 먹이가 있으면서 상어의 사이즈보다 작다면?
map[cur[0]][cur[1]] = 0; // 물고기를 제거
eat++;
move += cur[2]; // 움직인 거리를 총 움직인 거리에 추가
ck = true; // 먹이 먹었다고 체크
break;
}
for (int k = 0; k < 4; k++) {
int ny = cur[0] + dy[k];
int nx = cur[1] + dx[k];
if (ny < 0 || nx < 0 || nx >= N || ny >= N || visit[ny][nx] || map[ny][nx] > size)
continue;
que.add(new int[]{ny, nx, cur[2] + 1});
visit[ny][nx] = true;
}
}
if (!ck) // 큐가 비워질 때까지 먹이를 먹은적이 없다면, 더 이상 먹은 물고기가 없으므로 탈출
break;
if (size == eat) { // 사이즈와 먹이를 먹은 수가 동일하다면 상어의 크기를 증가
size++;
eat = 0;
}
}
System.out.println(move);
}
}
728x90
'[개발관련] > 코테준비' 카테고리의 다른 글
[백준] 연속합 (0) | 2021.07.18 |
---|---|
[백준] 공유기 설치 (0) | 2021.07.18 |
[백준] 출석체크 (0) | 2021.07.11 |
[백준] 징검다리 건너기 (0) | 2021.07.11 |
[백준] 공주님을 구해라! (0) | 2021.07.11 |