코드
import java.util.*;
class Node{
int data;
Node lt, rt;
public Node(int val) {
data=val;
lt=rt=null;
}
}
public class Tree말단노드까지의가장짧은경로BFS{
Node root;
public int BFS2(Node root){
Queue<Node> Q=new LinkedList<>();
Q.offer(root);
int L = 0 ;
while(!Q.isEmpty()) {
int len = Q.size();
for (int i = 0; i < len; i++) {
Node cur = Q.poll();
if(cur.lt ==null && cur.rt == null) return L;//자식노드가 없다면 말단
if(cur.lt != null) Q.add(cur.lt);
if(cur.rt != null) Q.add(cur.rt);
}
L++;
}
return 0;
}
public int BFS(Node root){
Queue<Node> Q=new LinkedList<>();
Q.offer(root);
int L=0;
while(!Q.isEmpty()){
int len=Q.size();
for(int i=0; i<len; i++){
Node cur=Q.poll();
if(cur.lt==null && cur.rt==null) return L;
if(cur.lt!=null) Q.offer(cur.lt);
if(cur.rt!=null) Q.offer(cur.rt);
}
L++;
}
return 0;
}
public static void main(String args[]) {
Tree말단노드까지의가장짧은경로BFS tree=new Tree말단노드까지의가장짧은경로BFS();
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.BFS(tree.root));
}
}
결과
1
자식 노드가 없으면 당연히 말단이다. 그리고 BFS탐색 특성상 가장 먼저 발견한 것이 가장 최단 거리이므로 바로 리턴한다.
'자료구조&알고리즘 > 자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비' 카테고리의 다른 글
그래프최단거리 (0) | 2023.03.03 |
---|---|
경로탐색 인접리스트 (0) | 2023.02.28 |
BFS상태트리_송아지찾기 (0) | 2023.02.24 |
레벨탐색 BFS (0) | 2023.02.22 |
부분 집합 구하기 (0) | 2023.02.20 |