inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

2. 바둑이 승차

바둑이 승차 - bfs 시간초과

261

김주영

작성한 질문수 6

0

 

 

안녕하세요, 바둑이 승차를 bfs로 풀어보다가, 시간 초과 에러가 떠서 여쭤봅니다.!

아래는 제 코드입니다.

import java.util.*;

public class Main {
    static class Node {
        int sum;
        int level;
        public Node(int sum, int level) {
            this.sum = sum;
            this.level = level;
        }
    }

    static int n;
    static int w;
    static int[] puppies;
    static int result;

    public int bfs() {
        Queue<Integer> Q = new LinkedList<>();
        Q.offer(puppies[0]);
        Q.offer(0);
        int L = 0;
        while(!Q.isEmpty()) {
            int len = Q.size();
            if(L==n) break;
            for(int i=0 ; i<len; i++) {
                int tmp = Q.poll();
                if(tmp>w) continue;
                result = Math.max(tmp, result);
                if(L<n-1 ) {
                    int next = puppies[L+1];
                    Q.offer(tmp + next);
                    Q.offer(tmp);
                }
            }
            L++;
        }
        return result;
    }

    public static void main(String[] args) {
        Main T = new Main();
        Scanner sc = new Scanner(System.in);
        w = sc.nextInt();
        n = sc.nextInt();
        puppies = new int[n];
        for(int i=0; i<n; i++) {
            puppies[i] = sc.nextInt();
        }
        T.bfs();
        System.out.println(result);
    }
}

 

bfs, dfs 모두 모든 노드를 방문한다는 점에서, 같을 것 같은데, 왜 bfs는 시간 초과가 날까요.?

제가 break을 놓친 부분이 있을까요?

 

 

 

 

java 코딩-테스트

답변 1

0

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다.

님의 바둑이 승차 문제를 BFS로 접근하셨을 때 시간 초과가 발생하는 이유는 BFS 알고리즘이 각 레벨에서 모든 경우의 수를 큐에 넣기 때문입니다. 이 경우 가지치기(pruning)나 다른 최적화 방법 없이는 탐색해야 할 상태 공간이 기하급수적으로 증가하여 시간 초과가 발생할 수 있습니다.

주어진 코드에서는, 각 레벨이 바둑이 한 마리에 해당하므로, 최악의 경우 (2^n) 개의 상태를 큐에 넣을 수 있고, 이로 인해 시간이 매우 오래 걸릴 수 있습니다. 특히 w가 클 경우에는 상당한 시간이 소요될 것입니다.

문제 해결을 위한 몇 가지 제안을 드리자면:

  1. 가지치기(pruning): 과정 중에 무게의 합이 이미 w를 초과한 경우 더 이상 탐색하지 않도록 합니다.
  2. 메모이제이션(memoization): 같은 레벨의 같은 무게의 합을 가진 경우를 저장하여 중복 계산을 방지합니다.
  3. BFS 대신 DFS 사용: DFS는 재귀적으로 각 경우를 탐색하고, 가지치기를 더 쉽게 적용할 수 있어, 이 문제에서 조금 더 적합할 수 있습니다.

아래의 코드는 시간 초과 문제를 해결하기 위해 필요한 수정 사항을 보여주고 있지는 않지만, 위에서 제안한 방법을 적용하여 최적화를 시도해 보시길 추천드립니다.

안녕하세요. 바뀐 채점사이트 관련해서 문의드립니다.

0

27

1

갑자기 채점 사이트가 바뀌었어요

0

32

1

문제 리스트 페이지

0

27

1

채점 사이트 관련 질문드립니다

0

23

1

봉우리 문제 질문입니다

0

81

2

씨름 선수 문제에서 각 선수의 몸무게나 키가 같을 수도 있다면?

0

64

0

이 코드랑 영상 코드중에 뭐가 더 좋은 코드인가요?

0

72

0

가중치 방향 그래프에서 가중치가 0인 간선을 표현하는 방법

0

67

1

좌표 정렬 문제 이 코드가 왜 틀린지 모르겠습니다 ㅠㅠ

0

85

2

6-7 강의에서

0

48

1

6-6. 장난꾸러기 질문 있습니다.

0

45

1

강의 수강후 코딩테스트

0

109

1

answer 변수 사용 여부

0

44

1

2중 for문

1

85

2

2-11. 임시반장정하기 (Runtime Error)

0

63

1

혹시 LinkedList 같은 자료 구조들은 따로 배우지 않나요?

0

70

1

이런 풀이는 어떨까요

0

43

1

자바 스트림 방식의 효율성 질문 드립니다.

0

57

1

알고리즘 자료 구조들..

0

62

1

StringBuilder vs BufferdWriter

0

47

1

원더랜드(프림)

0

50

1

이런 코드는 어떤가요?

0

61

1

bfs 풀이

0

57

1

병합정렬

0

56

1