mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-07 05:04:20 +00:00
Yudong Jin
772183705e
* Add Russian docs site baseline * Add Russian localized codebase * Polish Russian code wording * Update ru code translation. * Update code translation and chapter covers. * Fix pythontutor extraction. * Add README and landing page. * placeholder of profiles * Use figures of English version * Remove chapter paperbook
38 lines
1.5 KiB
Go
38 lines
1.5 KiB
Go
// File: build_tree.go
|
|
// Created Time: 2023-07-20
|
|
// Author: hongyun-robot (1836017030@qq.com)
|
|
|
|
package chapter_divide_and_conquer
|
|
|
|
import . "github.com/krahets/hello-algo/pkg"
|
|
|
|
/* Построить двоичное дерево: разделяй и властвуй */
|
|
func dfsBuildTree(preorder []int, inorderMap map[int]int, i, l, r int) *TreeNode {
|
|
// Завершить при пустом диапазоне поддерева
|
|
if r-l < 0 {
|
|
return nil
|
|
}
|
|
// Инициализировать корневой узел
|
|
root := NewTreeNode(preorder[i])
|
|
// Найти m, чтобы разделить левое и правое поддеревья
|
|
m := inorderMap[preorder[i]]
|
|
// Подзадача: построить левое поддерево
|
|
root.Left = dfsBuildTree(preorder, inorderMap, i+1, l, m-1)
|
|
// Подзадача: построить правое поддерево
|
|
root.Right = dfsBuildTree(preorder, inorderMap, i+1+m-l, m+1, r)
|
|
// Вернуть корневой узел
|
|
return root
|
|
}
|
|
|
|
/* Построить двоичное дерево */
|
|
func buildTree(preorder, inorder []int) *TreeNode {
|
|
// Инициализировать хеш-таблицу для хранения соответствия элементов inorder их индексам
|
|
inorderMap := make(map[int]int, len(inorder))
|
|
for i := 0; i < len(inorder); i++ {
|
|
inorderMap[inorder[i]] = i
|
|
}
|
|
|
|
root := dfsBuildTree(preorder, inorderMap, 0, 0, len(inorder)-1)
|
|
return root
|
|
}
|