题目概述
- 算法说明
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。 - 测试用例
输入:
10
/ \
6 14
/ \ / \
4 8 12 16
输出:
4<=>6<=>8<=>10<=>12<=>14<=>16
解析&参考答案
- 解析
方法1: 将节点全部存放在数组中,然后再从左到右组装一个链表即可
方法2: 使用中序遍历,新增pre和head节点,调换节点的位置来链接出一个新的链表 - 参考答案
vim jz26.go
package main
import "fmt"
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
var pre, head *TreeNode
func ConvertCore(root *TreeNode) {
if root == nil {
return
}
ConvertCore(root.Left)
if pre == nil {
head = root
} else {
pre.Right = root
root.Left = pre
}
pre = root
ConvertCore(root.Right)
}
func Convert(pRootOfTree *TreeNode) *TreeNode {
if pRootOfTree == nil {
return nil
}
pre, head = nil, nil
ConvertCore(pRootOfTree)
return head
}
func PrintList(head *TreeNode) {
tmp := head
for tmp != nil {
fmt.Printf("%d->", tmp.Val)
tmp = tmp.Right
}
}
func main() {
root := &TreeNode{10, &TreeNode{5, &TreeNode{4, nil, nil}, &TreeNode{7, nil, nil}},
&TreeNode{12, nil, nil}}
result := Convert(root)
PrintList(result)
}
注意事项
- to add
说明
- 当前使用 go1.15.8
- 参考 牛客网--剑指offer
标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
- 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解