inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

14. 그래프 최단거리(BFS)

숙제 검사

185

jay

작성한 질문수 4

0

안녕하세요 선생님
숙제로 내주신 문제를 풀어보았습니다.

좀 무식한 방법으로 한 것 같은데 이것 말고는 다른 방법이 떠오르지가 않더라구요.

피드백 부탁드립니다!


public class Main {

    static int[] visited;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int nodeCount = Integer.parseInt(st.nextToken()); // 노드의 수
        int lineCount = Integer.parseInt(st.nextToken()); // 간선의 수

        List<Integer>[] list = new ArrayList[nodeCount + 1];
        for (int i = 1; i <= nodeCount; i++) {
            list[i] = new ArrayList<>();
        }
        visited = new int[nodeCount + 1];

        for (int i = 0; i < lineCount; i++) {
            st = new StringTokenizer(br.readLine());
            int nodeA = Integer.parseInt(st.nextToken());
            int nodeB = Integer.parseInt(st.nextToken());

            list[nodeA].add(nodeB);
        }

        bfs(list, 2);
    }

    private static void bfs(List<Integer>[] list, int target) {
        Queue<List<Integer>> q = new LinkedList<>();
        q.offer(list[1]);
        visited[1] = 1;

        int level = 1;
        while (!q.isEmpty()) {
            boolean isFinished = false;
            int size = q.size();

            loopOut:
            for (int i = 0; i < size; i++) {
                List<Integer> currentNodes = q.poll();
                for (Integer nextNode : currentNodes) {
                    if (nextNode == target) {
                        System.out.println(target + " : " + level);
                        isFinished = true;
                        visited[nextNode] = 0;
                        break loopOut;
                    } else {
                        List<Integer> e = list[nextNode];
                        if (!e.isEmpty() && visited[nextNode] == 0) {
                            visited[nextNode] = 1;
                            q.offer(e);
                        }
                    }
                }
            }

            if (!isFinished) {
                level++;
            } else {
                target++;
                level = 1;
                q.clear();
                q.offer(list[1]);
                for (int i = 2; i < visited.length; i++) {
                    visited[i] = 0;
                }
                visited[1] = 1;
            }
        }
    }
}

java 코딩-테스트

답변 2

0

김태원

안녕하세요^^

잘 하셨습니다. 코드가 약간 복잡해 보이네요. 아래 인프런 AI 인턴이 답한 다른 분의 코드를 참조하면 좋을 것 같습니다.

0

인프런 AI 인턴

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

작성해 주신 코드를 보니, 그래프를 탐색하여 특정 목표 노드를 찾는 BFS(Breadth-First Search) 방식으로 잘 구현해 주셨습니다. 다만, 좀 더 효율적이고 깔끔하게 구현할 수 있는 방법들이 있습니다. 참고로, 다른 학습자분들의 유사한 질문과 답변 링크를 통해 추가적인 팁을 얻으실 수 있습니다.

  1. 숙제 코드 리뷰 부탁드립니다

    • 이 질문에서는 BFS와 인접리스트를 통해 그래프 탐색을 구현한 코드에 대한 리뷰가 포함되어 있습니다. 유사한 BFS 알고리즘을 다루고 있으므로 코드 작성에 도움이 될 것입니다.
  2. 숙제검사

    • 여기에서도 Java로 그래프의 최단거리를 BFS로 구하는 예제 코드가 리뷰되었습니다. 특히, 코드 구조와 BFS 구현에 대해 참고하실 수 있습니다.

기본적으로 현재 코드를 개선할 수 있는 몇 가지 팁을 드리자면:
- loopOut 레이블을 제거하고 코드 흐름을 단순화할 수 있습니다.
- BFS의 기본 아이디어는 큐를 사용하여 각 레벨의 노드를 탐색하는 것이므로, 노드 리스트를 큐에 넣고 탐색하는 것으로 충분합니다.
- visited 배열은 boolean 타입으로 해주시면 더 간결해집니다.

최종적으로 문제 해결을 위한 최적화된 방법은 여러 코드를 참고하면서 점진적으로 개선하는 것이 좋습니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다.
현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏
추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.

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

0

33

1

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

0

34

1

문제 리스트 페이지

0

29

1

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

0

24

1

봉우리 문제 질문입니다

0

84

2

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

0

65

0

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

0

72

0

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

0

67

1

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

0

85

2

6-7 강의에서

0

48

1

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

0

46

1

강의 수강후 코딩테스트

0

111

1

answer 변수 사용 여부

0

46

1

2중 for문

1

85

2

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

0

63

1

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

0

70

1

이런 풀이는 어떨까요

0

44

1

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

0

57

1

알고리즘 자료 구조들..

0

63

1

StringBuilder vs BufferdWriter

0

48

1

원더랜드(프림)

0

50

1

이런 코드는 어떤가요?

0

61

1

bfs 풀이

0

57

1

병합정렬

0

57

1