인프런 커뮤니티 질문&답변

dmsgml9903님의 프로필 이미지
dmsgml9903

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

8. 송아지 찾기1(BFS)

송아지 찾기 시간초과 질문

작성

·

130

·

수정됨

0

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

public class FindCow {

    static int personPos;
    static int cowPos;
    static int[] idx = {5, 1, -1};

    public static int solution(int n) {
        boolean[] visit = new boolean[10_001];
        Queue<Integer> queue = new LinkedList<>();
        visit[n] = true;

        queue.offer(n);
        int level = 0;
        while (!queue.isEmpty()) {
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                Integer curPos = queue.poll();
                for (int k = 0; k < 3; k++) {
                    int nextPos = curPos + idx[k];
                    if (nextPos == cowPos) {
                        return level + 1;
                    }
                    if (nextPos < 10_001 && nextPos > 0 && !visit[nextPos]) {
                        queue.offer(nextPos);
                        visit[nextPos] = false;
                    }
                }
            }
            level++;
        }
        return 1;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        personPos = in.nextInt();
        cowPos = in.nextInt();
        System.out.println(solution(personPos));
    }
}

안녕하세요! 좋은 강의 항상 감사드립니다.

강의 중, 송아지 찾기(BFS)에서 시간 초과가 떠서 확인해 보니, 강의에서 올려주신 코드와 딱히 다른 부분이 없어 보이는데 시간초과가 납니다. 왜 그럴까요?

스크린샷 2024-02-26 오전 9.16.30.png

답변 1

0

안녕하세요, 인프런 AI 인턴입니다.
dmsgml9903님, BFS 알고리즘 관련 질문 감사합니다.

해당 코드에서 시간 초과가 발생하는 이유는 방문 처리 로직에 오류가 있어서 무한 루프에 빠지거나 같은 정점을 계속 방문하게 되는 비효율적인 검색이 진행되기 때문인 것 같습니다.

visit[nextPos] = false; 부분에 문제가 있습니다. 이미 방문한 위치를 true로 바꾸어야 하는데, false로 설정하고 있습니다. 따라서 이미 방문한 위치를 또 방문하는 상황이 발생합니다. 그 부분을 visit[nextPos] = true;로 바꿔주면 에러를 해결할 수 있을 것입니다.

또한, 문제 해결에 도움이 되는 이미지를 링크로 첨부하오니 참고해 주세요.
재채점 결과 보기

해당 코드를 수정하여 다시 실행해보시고, 문제가 지속될 경우 다시 문의해 주세요.

dmsgml9903님의 프로필 이미지
dmsgml9903

작성한 질문수

질문하기