Algorithm

[Algorithm/Java] 알고리즘 자바 이진트리 최단거리 DFS (코딩테스트, Recursive, Tree, Graph, DFS, BFS, Recursion Function, 말단 노드까지의 최단거리)

권락현 2022. 5. 10. 00:00


1. 문제 설명

  • 문제
    Tree 말단 노드까지의 가장 짧은 경로
    이진트리에서 루트 노드 1에서 말단 노드까지의 길이 중 가장 짧은 길이를 구하시오.
    각 경로의 길이는 간선의 개수로 한다.
  • 출력
    말단 노드까지의 길이 중 가장 짧은 길이를 출력한다.

 

 

2. 문제 코드 및 풀이 설명

// 이진 트리의 노드 클래스
class Node {
  int data;
  Node lt, rt;

  public Node(int val) {
    data = val;
    lt = rt = null;
  }
}


public class Tree_DFS {
  Node root;

  public static void main(String[] args) {
    Tree_DFS tree = new Tree_DFS();
    tree.root = new Node(1);
    tree.root.lt = new Node(2);
    tree.root.rt = new Node(3);
    tree.root.lt.lt = new Node(4);
    tree.root.lt.rt = new Node(5);
    System.out.println(tree.DFS(0, tree.root));
  }

  // DFS 최단 거리 알고리즘
  public int DFS(int L, Node root) {
    // 자식 노드가 없는 노드까지의 거리 중 가장 짧은 거리를 리턴
    if (root.lt == null && root.rt == null) return L; // 만약 자식 노드가 없다면 리턴
    else return Math.min(DFS(L + 1, root.lt), DFS(L + 1, root.rt)); // 자식 노드 중 작은 값을 리턴

  }
}

 

3. 출력 예시

출력
1

 

 

 


위의 내용은 인프런에서 수강할 수 있는 김태원님의 자바 알고리즘 문제풀이 강의를 바탕으로 공부한 내용을 정리한 내용입니다!

https://inf.run/iAi6

 

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의

자바(Java)로 코딩테스트를 준비하시는 분을 위한 강좌입니다. 코딩테스트에서 가장 많이 출제되는 Top 10 Topic을 다루고 있습니다. 주제와 연동하여 기초문제부터 중급문제까지 단계적으로 구성

www.inflearn.com