mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-11 00:11:00 +08:00
7257641e0e
* Retranslate the complete Japanese online edition from Chinese * Fit Japanese homepage labels and localize heading permalinks * Point Japanese contribution instructions to the Japanese preview * Replace locale conditionals with multilingual UI tables
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
// File: tree_node.go
|
|
// Created Time: 2022-11-25
|
|
// Author: Reanon (793584285@qq.com)
|
|
|
|
package pkg
|
|
|
|
// TreeNode は二分木のノード
|
|
type TreeNode struct {
|
|
Val any // ノードの値
|
|
Height int // ノードの高さ
|
|
Left *TreeNode // 左の子ノードへの参照
|
|
Right *TreeNode // 右の子ノードへの参照
|
|
}
|
|
|
|
// NewTreeNode は二分木のノードのコンストラクタ
|
|
func NewTreeNode(v any) *TreeNode {
|
|
return &TreeNode{
|
|
Val: v,
|
|
Height: 0,
|
|
Left: nil,
|
|
Right: nil,
|
|
}
|
|
}
|
|
|
|
// シリアライズの符号化規則は次を参照：
|
|
// https://www.hello-algo.com/ja/chapter_tree/array_representation_of_tree/
|
|
// 二分木の配列表現：
|
|
// [1, 2, 3, 4, nil, 6, 7, 8, 9, nil, nil, 12, nil, nil, 15]
|
|
// 二分木の連結リスト表現：
|
|
//
|
|
// /——— 15
|
|
// /——— 7
|
|
// /——— 3
|
|
// | \——— 6
|
|
// | \——— 12
|
|
//
|
|
// ——— 1
|
|
//
|
|
// \——— 2
|
|
// | /——— 9
|
|
// \——— 4
|
|
// \——— 8
|
|
|
|
// SliceToTreeDFS はリストをデシリアライズして二分木にする：再帰
|
|
func SliceToTreeDFS(arr []any, i int) *TreeNode {
|
|
if i < 0 || i >= len(arr) || arr[i] == nil {
|
|
return nil
|
|
}
|
|
root := NewTreeNode(arr[i])
|
|
root.Left = SliceToTreeDFS(arr, 2*i+1)
|
|
root.Right = SliceToTreeDFS(arr, 2*i+2)
|
|
return root
|
|
}
|
|
|
|
// SliceToTree はスライスをデシリアライズして二分木にする
|
|
func SliceToTree(arr []any) *TreeNode {
|
|
return SliceToTreeDFS(arr, 0)
|
|
}
|
|
|
|
// TreeToSliceDFS は二分木をシリアライズしてスライスにする：再帰
|
|
func TreeToSliceDFS(root *TreeNode, i int, res *[]any) {
|
|
if root == nil {
|
|
return
|
|
}
|
|
for i >= len(*res) {
|
|
*res = append(*res, nil)
|
|
}
|
|
(*res)[i] = root.Val
|
|
TreeToSliceDFS(root.Left, 2*i+1, res)
|
|
TreeToSliceDFS(root.Right, 2*i+2, res)
|
|
}
|
|
|
|
// TreeToSlice は二分木をシリアライズしてスライスにする
|
|
func TreeToSlice(root *TreeNode) []any {
|
|
var res []any
|
|
TreeToSliceDFS(root, 0, &res)
|
|
return res
|
|
}
|