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
72 lines
2.3 KiB
Swift
72 lines
2.3 KiB
Swift
/**
|
||
* File: TreeNode.swift
|
||
* Created Time: 2023-01-02
|
||
* Author: nuomi1 (nuomi1@qq.com)
|
||
*/
|
||
|
||
/* 二分木のノードクラス */
|
||
public class TreeNode {
|
||
public var val: Int // ノードの値
|
||
public var height: Int // ノードの高さ
|
||
public var left: TreeNode? // 左の子ノードへの参照
|
||
public var right: TreeNode? // 右の子ノードへの参照
|
||
|
||
/* コンストラクタ */
|
||
public init(x: Int) {
|
||
val = x
|
||
height = 0
|
||
}
|
||
|
||
// シリアライズの符号化規則は次を参照：
|
||
// 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
|
||
|
||
/* リストから二分木へデシリアライズ：再帰 */
|
||
private static func listToTreeDFS(arr: [Int?], i: Int) -> TreeNode? {
|
||
if i < 0 || i >= arr.count || arr[i] == nil {
|
||
return nil
|
||
}
|
||
let root = TreeNode(x: arr[i]!)
|
||
root.left = listToTreeDFS(arr: arr, i: 2 * i + 1)
|
||
root.right = listToTreeDFS(arr: arr, i: 2 * i + 2)
|
||
return root
|
||
}
|
||
|
||
/* リストから二分木へデシリアライズ */
|
||
public static func listToTree(arr: [Int?]) -> TreeNode? {
|
||
listToTreeDFS(arr: arr, i: 0)
|
||
}
|
||
|
||
/* 二分木をシリアライズしてリストにする：再帰 */
|
||
private static func treeToListDFS(root: TreeNode?, i: Int, res: inout [Int?]) {
|
||
if root == nil {
|
||
return
|
||
}
|
||
while i >= res.count {
|
||
res.append(nil)
|
||
}
|
||
res[i] = root?.val
|
||
treeToListDFS(root: root?.left, i: 2 * i + 1, res: &res)
|
||
treeToListDFS(root: root?.right, i: 2 * i + 2, res: &res)
|
||
}
|
||
|
||
/* 二分木をシリアライズしてリストにする */
|
||
public static func treeToList(root: TreeNode?) -> [Int?] {
|
||
var res: [Int?] = []
|
||
treeToListDFS(root: root, i: 0, res: &res)
|
||
return res
|
||
}
|
||
}
|