mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-10 04:06:51 +08:00
Update 二叉树的迭代遍历.md
添加 二叉树的迭代遍历 GO版本
This commit is contained in:
@ -163,7 +163,109 @@ Python:
|
|||||||
|
|
||||||
|
|
||||||
Go:
|
Go:
|
||||||
|
> 迭代法前序遍历
|
||||||
|
|
||||||
|
```go
|
||||||
|
//迭代法前序遍历
|
||||||
|
/**
|
||||||
|
type Element struct {
|
||||||
|
// 元素保管的值
|
||||||
|
Value interface{}
|
||||||
|
// 内含隐藏或非导出字段
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *List) Back() *Element
|
||||||
|
前序遍历:中左右
|
||||||
|
压栈顺序:右左中
|
||||||
|
**/
|
||||||
|
func preorderTraversal(root *TreeNode) []int {
|
||||||
|
if root == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var stack = list.New()
|
||||||
|
stack.PushBack(root.Right)
|
||||||
|
stack.PushBack(root.Left)
|
||||||
|
res:=[]int{}
|
||||||
|
res=append(res,root.Val)
|
||||||
|
for stack.Len()>0 {
|
||||||
|
e:=stack.Back()
|
||||||
|
stack.Remove(e)
|
||||||
|
node := e.Value.(*TreeNode)//e是Element类型,其值为e.Value.由于Value为接口,所以要断言
|
||||||
|
if node==nil{
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
res=append(res,node.Val)
|
||||||
|
stack.PushBack(node.Right)
|
||||||
|
stack.PushBack(node.Left)
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
> 迭代法后序遍历
|
||||||
|
|
||||||
|
```go
|
||||||
|
//后续遍历:左右中
|
||||||
|
//压栈顺序:中右左(按照前序遍历思路),再反转结果数组
|
||||||
|
func postorderTraversal(root *TreeNode) []int {
|
||||||
|
if root == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var stack = list.New()
|
||||||
|
stack.PushBack(root.Left)
|
||||||
|
stack.PushBack(root.Right)
|
||||||
|
res:=[]int{}
|
||||||
|
res=append(res,root.Val)
|
||||||
|
for stack.Len()>0 {
|
||||||
|
e:=stack.Back()
|
||||||
|
stack.Remove(e)
|
||||||
|
node := e.Value.(*TreeNode)//e是Element类型,其值为e.Value.由于Value为接口,所以要断言
|
||||||
|
if node==nil{
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
res=append(res,node.Val)
|
||||||
|
stack.PushBack(node.Left)
|
||||||
|
stack.PushBack(node.Right)
|
||||||
|
}
|
||||||
|
for i:=0;i<len(res)/2;i++{
|
||||||
|
res[i],res[len(res)-i-1] = res[len(res)-i-1],res[i]
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
> 迭代法中序遍历
|
||||||
|
|
||||||
|
```go
|
||||||
|
//迭代法中序遍历
|
||||||
|
func inorderTraversal(root *TreeNode) []int {
|
||||||
|
rootRes:=[]int{}
|
||||||
|
if root==nil{
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
stack:=list.New()
|
||||||
|
node:=root
|
||||||
|
//先将所有左节点找到,加入栈中
|
||||||
|
for node!=nil{
|
||||||
|
stack.PushBack(node)
|
||||||
|
node=node.Left
|
||||||
|
}
|
||||||
|
//其次对栈中的每个节点先弹出加入到结果集中,再找到该节点的右节点的所有左节点加入栈中
|
||||||
|
for stack.Len()>0{
|
||||||
|
e:=stack.Back()
|
||||||
|
node:=e.Value.(*TreeNode)
|
||||||
|
stack.Remove(e)
|
||||||
|
//找到该节点的右节点,再搜索他的所有左节点加入栈中
|
||||||
|
rootRes=append(rootRes,node.Val)
|
||||||
|
node=node.Right
|
||||||
|
for node!=nil{
|
||||||
|
stack.PushBack(node)
|
||||||
|
node=node.Left
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return rootRes
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user