글 읽기 - 메모리 초과

rds257 3년 전 0

안녕하세요. 코딩 초보입니다. 질문 하나만 하겠습니다.

일단 테케는 통과 했는데, 제출 할 경우 메모리 초과됩니다.

어떤 부분을 수정 하면 좋을까요?

답변 부탁드립니다.

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Main {

	private static int[][][] map;
	private static int[] addRow = { 1, -1, 0, 0, 0, 0 };
	private static int[] addCol = { 0, 0, 1, -1, 0, 0 };
	private static int[] addFac = { 0, 0, 0, 0, 1, -1 };
	private static int n, m, h;
	private static int max = 0;

	private static Queue<int[]> queue = new LinkedList<int[]>();

	public static void main(String[] args) {

		Scanner sc = new Scanner(System.in);
		m = sc.nextInt();
		n = sc.nextInt();
		h = sc.nextInt();
		map = new int[h][n][m];
		for (int i = 0; i < h; i++) {
			for (int j = 0; j < n; j++) {
				for (int j2 = 0; j2 < m; j2++) {
					int a = sc.nextInt();
					map[i][j][j2] = a;
					if (a == 1) {
						queue.add(new int[] { i, j, j2 });
					}
				}
			}
		}

		bfs(queue);

		if (!check()) {
			System.out.println(-1);
		} else if (max == 0) {
			System.out.println(0);
		} else {
			System.out.println(max - 1);
		}
	}

	public static void bfs(Queue<int[]> queue) {

		while (!queue.isEmpty()) {
			int c = queue.peek()[0];
			int a = queue.peek()[1];
			int b = queue.peek()[2];
			queue.poll();
			for (int i = 0; i < 6; i++) {
				int nz = c + addFac[i];
				int nx = a + addRow[i];
				int ny = b + addCol[i];

				if (nx >= 0 && ny >= 0 && nz >= 0 && nx < n && ny < m && nz < h) {
					if (map[nz][nx][ny] == 0) {
						map[nz][nx][ny] = map[c][a][b] + 1;
						queue.add(new int[] { nz, nx, ny });
						if (map[nz][nx][ny] > max) {
							max = map[nz][nx][ny];
						}
					}
				}
			}

		}
	}

	public static boolean check() {
		for (int i = 0; i < h; i++) {
			for (int j = 0; j < n; j++) {
				for (int j2 = 0; j2 < m; j2++) {
					if (map[i][j][j2] == 0) {
						return false;
					}
				}
			}
		}
		return true;
	}
}

rlarla97 3년 전 0

Scanner 말고 BufferedReader을 사용하시길 바랍니다. Scanner가 입력스트림에서 문자열을 받아오면 Scanner 내부에서 정규식 검사를 하는데 그 비용이 생각이상으로 비쌉니다. 아마 BufferedReader로만 바꾸어주어도 해결 가능할 것 같습니다.

rds257 3년 전 0

rlarla97님 감사합니다

덕분에 배워갑니다^^

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

메모리 초과

rds257 3년 전 0

rlarla97 3년 전 0

rds257 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말