mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-02 03:05:24 +08:00
Yudong Jin
2f561d3d51
* feat(ko): add the complete Korean online edition * Use UTF-8 for Python CI output on Windows * Remove print-book promotion from the Korean site * fix(ko): preserve emphasis before Korean particles
42 lines
1.3 KiB
Kotlin
42 lines
1.3 KiB
Kotlin
/**
|
|
* File: binary_tree_bfs.kt
|
|
* Created Time: 2024-01-25
|
|
* Author: curtishd ([email protected])
|
|
*/
|
|
|
|
package chapter_tree
|
|
|
|
import utils.TreeNode
|
|
import utils.printTree
|
|
import java.util.*
|
|
|
|
/* 레벨 순서 순회 */
|
|
fun levelOrder(root: TreeNode?): MutableList<Int> {
|
|
// 큐를 초기화하고 루트 노드를 넣는다
|
|
val queue = LinkedList<TreeNode?>()
|
|
queue.add(root)
|
|
// 순회 순서를 저장할 리스트를 초기화한다
|
|
val list = mutableListOf<Int>()
|
|
while (queue.isNotEmpty()) {
|
|
val node = queue.poll() // 큐에서 꺼낸다
|
|
list.add(node?._val!!) // 노드 값을 저장한다
|
|
if (node.left != null)
|
|
queue.offer(node.left) // 왼쪽 자식 노드를 큐에 삽입한다
|
|
if (node.right != null)
|
|
queue.offer(node.right) // 오른쪽 자식 노드를 큐에 삽입한다
|
|
}
|
|
return list
|
|
}
|
|
|
|
/* 실행 코드 */
|
|
fun main() {
|
|
/* 이진 트리 초기화 */
|
|
// 여기서는 리스트로 이진 트리를 바로 생성하는 함수를 사용한다
|
|
val root = TreeNode.listToTree(mutableListOf(1, 2, 3, 4, 5, 6, 7))
|
|
println("\n이진 트리 초기화\n")
|
|
printTree(root)
|
|
|
|
/* 레벨 순서 순회 */
|
|
val list = levelOrder(root)
|
|
println("\n레벨 순서 순회의 노드 출력 순서 = $list")
|
|
} |