分治03--二叉搜索树和双向链表

题目概述

• 算法说明
  输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。
• 测试用例
  输入：
    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)
}

注意事项

1. to add

说明

1. 当前使用 go1.15.8
2. 参考 牛客网--剑指offer
   标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解